Home
last modified time | relevance | path

Searched defs:operand_array (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator_test.cc3015 auto operand_array = std::make_unique<Array2D<float>>(3, 5); in TEST_P() local
3048 auto operand_array = std::make_unique<Array2D<float>>(2, 4); in TEST_P() local
3086 auto operand_array = std::make_unique<Array2D<float>>(2, 4); in TEST_P() local
3122 auto operand_array = std::make_unique<Array2D<double>>(2, 3); in TEST_P() local
3161 auto operand_array = std::make_unique<Array2D<double>>(2, 3); in TEST_P() local
3197 auto operand_array = std::make_unique<Array2D<double>>(2, 3); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
H A Dllvm_loop.cc259 const llvm_ir::IrArray& operand_array, int64_t dimension_to_skip, in EmitOperandArrayLoopNest()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc802 llvm_ir::IrArray operand_array(GetIrArrayFor(operand)); in HandleSelectAndScatter() local
2147 llvm_ir::IrArray operand_array(GetIrArrayFor(operand)); in HandlePad() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.cc1944 const IrArray& operand_array = ir_arrays[0]; in EmitSelectAndScatter() local