Searched refs:operand_array (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_evaluator_test.cc | 1774 auto operand_array = absl::make_unique<Array2D<float>>(3, 5); in TEST_P() local 1775 operand_array->FillUnique(1.0f); in TEST_P() 1777 LiteralUtil::CreateR2FromArray2D<float>(*operand_array); in TEST_P() 1807 auto operand_array = absl::make_unique<Array2D<float>>(2, 4); in TEST_P() local 1808 operand_array->FillUnique(1.0f); in TEST_P() 1810 LiteralUtil::CreateR2FromArray2D<float>(*operand_array); in TEST_P() 1845 auto operand_array = absl::make_unique<Array2D<float>>(2, 4); in TEST_P() local 1846 operand_array->FillUnique(1.0f); in TEST_P() 1848 LiteralUtil::CreateR2FromArray2D<float>(*operand_array); in TEST_P() 1881 auto operand_array = absl::make_unique<Array2D<double>>(2, 3); in TEST_P() local [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
D | llvm_loop.cc | 257 const llvm_ir::IrArray& operand_array, int64 dimension_to_skip, in EmitOperandArrayLoopNest() argument 263 const Shape& shape = operand_array.GetShape(); in EmitOperandArrayLoopNest()
|
D | llvm_loop.h | 254 const llvm_ir::IrArray& operand_array, int64 dimension_to_skip,
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | ir_emitter.cc | 856 llvm_ir::IrArray operand_array(GetIrArrayFor(operand)); in HandleSelectAndScatter() local 858 operand_multi_index, operand_array.GetShape(), b_.getInt64Ty()); in HandleSelectAndScatter() 860 operand_array.EmitReadArrayElement(operand_index, &b_); in HandleSelectAndScatter() 869 operand_array.EmitArrayElementAddress(operand_index, &b_); in HandleSelectAndScatter() 2122 llvm_ir::IrArray operand_array(GetIrArrayFor(operand)); in HandlePad() local 2124 operand_array.EmitReadArrayElement(operand_index, &b_); in HandlePad()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter_unnested.cc | 899 IrArray operand_array = GetIrArray(*operand, *select_and_scatter); in HandleSelectAndScatter() local 903 operand_array.EmitReadArrayElement(operand_index, &b_); in HandleSelectAndScatter() 913 operand_array.EmitArrayElementAddress(operand_index, &b_); in HandleSelectAndScatter()
|