Searched refs:operand_array (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_evaluator_test.cc | 3015 auto operand_array = std::make_unique<Array2D<float>>(3, 5); in TEST_P() local
3016 operand_array->FillUnique(1.0f); in TEST_P()
3018 LiteralUtil::CreateR2FromArray2D<float>(*operand_array); in TEST_P()
3048 auto operand_array = std::make_unique<Array2D<float>>(2, 4); in TEST_P() local
3049 operand_array->FillUnique(1.0f); in TEST_P()
3051 LiteralUtil::CreateR2FromArray2D<float>(*operand_array); in TEST_P()
3086 auto operand_array = std::make_unique<Array2D<float>>(2, 4); in TEST_P() local
3087 operand_array->FillUnique(1.0f); in TEST_P()
3089 LiteralUtil::CreateR2FromArray2D<float>(*operand_array); in TEST_P()
3122 auto operand_array = std::make_unique<Array2D<double>>(2, 3); in TEST_P() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| H A D | llvm_loop.cc | 259 const llvm_ir::IrArray& operand_array, int64_t dimension_to_skip, in EmitOperandArrayLoopNest() argument
265 const Shape& shape = operand_array.GetShape(); in EmitOperandArrayLoopNest()
|
| H A D | llvm_loop.h | 252 const llvm_ir::IrArray& operand_array, int64_t dimension_to_skip,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emitter.cc | 802 llvm_ir::IrArray operand_array(GetIrArrayFor(operand)); in HandleSelectAndScatter() local
804 operand_multi_index, operand_array.GetShape(), b_.getInt64Ty()); in HandleSelectAndScatter()
806 operand_array.EmitReadArrayElement(operand_index, &b_); in HandleSelectAndScatter()
815 operand_array.EmitArrayElementAddress(operand_index, &b_); in HandleSelectAndScatter()
817 Load(operand_array.GetElementLlvmType(), operand_address); in HandleSelectAndScatter()
833 Store(Load(operand_array.GetElementLlvmType(), operand_address), in HandleSelectAndScatter()
2147 llvm_ir::IrArray operand_array(GetIrArrayFor(operand)); in HandlePad() local
2149 operand_array.EmitReadArrayElement(operand_index, &b_); in HandlePad()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.cc | 1944 const IrArray& operand_array = ir_arrays[0]; in EmitSelectAndScatter() local
2062 operand_array.EmitReadArrayElement(operand_index, &b_); in EmitSelectAndScatter()
2072 operand_array.EmitArrayElementAddress(operand_index, &b_); in EmitSelectAndScatter()
2098 Store(Load(operand_array.GetElementLlvmType(), operand_address), in EmitSelectAndScatter()
|