Searched defs:rhs_array (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 208 const llvm_ir::IrArray& rhs_array, in DotOpEmitter()
847 const llvm_ir::IrArray& lhs_array, const llvm_ir::IrArray& rhs_array, in EmitNonBatchDotOperation()
930 const llvm_ir::IrArray& lhs_array, const llvm_ir::IrArray& rhs_array, in EmitBatchDotOperation()
1033 const llvm_ir::IrArray& rhs_array, in EmitDotOperation()
|
| D | ir_emitter.cc | 930 llvm_ir::IrArray rhs_array(GetIrArrayFor(rhs)); in HandleDot() local
2187 llvm_ir::IrArray rhs_array( in HandleFusion() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 746 bool CompareArrayBuffers(const Array& lhs_array, const Array& rhs_array) { in CompareArrayBuffers()
762 bool HaveSameMinMax(const Array& lhs_array, const Array& rhs_array) { in HaveSameMinMax()
775 const Array& rhs_array) { in HaveSameQuantizationParams()
789 bool CompareConstantArrays(const Array& lhs_array, const Array& rhs_array) { in CompareConstantArrays()
1304 const auto& rhs_array = *rhs_array_it->second; in DedupeConstantArrays() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 774 auto rhs_array = absl::make_unique<Array2D<float>>(3, 2); in TEST_P() local
818 auto rhs_array = absl::make_unique<Array2D<float>>(3, 2); in TEST_P() local
855 auto rhs_array = absl::make_unique<Array4D<float>>(2, 2, 3, 1); in TEST_P() local
899 Array3D<float> rhs_array = {{{3.f, 4.f}}}; in TEST_P() local
956 Array4D<float> rhs_array(1, 1, 2, 2); in TEST_P() local
1178 Array4D<float> rhs_array(1, 1, 2, 2); in TEST_P() local
1242 Array4D<float> rhs_array(1, 1, 2, 2); in TEST_P() local
1308 Array4D<float> rhs_array(1, 1, 2, 3); in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 472 const llvm_ir::IrArray& rhs_array = GetIrArray(*rhs_instruction, *dot); in HandleDot() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 1144 Array2D<float> rhs_array({{5.0f, 6.0f}, {7.0f, 8.0f}}); in XLA_TEST_F() local
|