Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_loop.cc258 const llvm_ir::IrArray& operand_array, int64 dimension_to_skip, in EmitOperandArrayLoopNest()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc2873 auto operand_array = absl::make_unique<Array2D<float>>(3, 5); in TEST_P() local
2906 auto operand_array = absl::make_unique<Array2D<float>>(2, 4); in TEST_P() local
2944 auto operand_array = absl::make_unique<Array2D<float>>(2, 4); in TEST_P() local
2980 auto operand_array = absl::make_unique<Array2D<double>>(2, 3); in TEST_P() local
3019 auto operand_array = absl::make_unique<Array2D<double>>(2, 3); in TEST_P() local
3055 auto operand_array = absl::make_unique<Array2D<double>>(2, 3); in TEST_P() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc916 llvm_ir::IrArray operand_array(GetIrArrayFor(operand)); in HandleSelectAndScatter() local
2365 llvm_ir::IrArray operand_array(GetIrArrayFor(operand)); in HandlePad() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc720 IrArray operand_array = GetIrArray(*operand, *select_and_scatter); in HandleSelectAndScatter() local