Searched defs:rhs_array (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 237 const llvm_ir::IrArray& lhs_array, const llvm_ir::IrArray& rhs_array, in DotOpEmitter()
1041 const llvm_ir::IrArray& lhs_array, const llvm_ir::IrArray& rhs_array, in EmitNonBatchDotOperation()
1126 const llvm_ir::IrArray& lhs_array, const llvm_ir::IrArray& rhs_array, in EmitBatchDotOperation()
1233 const llvm_ir::IrArray& rhs_array, in EmitDotOperation()
|
| D | ir_emitter.cc | 850 llvm_ir::IrArray rhs_array(GetIrArrayFor(rhs)); in HandleDot() local
2102 llvm_ir::IrArray rhs_array( in HandleFusion() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 774 bool CompareArrayBuffers(const Array& lhs_array, const Array& rhs_array) { in CompareArrayBuffers()
790 bool HaveSameMinMax(const Array& lhs_array, const Array& rhs_array) { in HaveSameMinMax()
803 const Array& rhs_array) { in HaveSameQuantizationParams()
817 bool CompareConstantArrays(const Array& lhs_array, const Array& rhs_array) { in CompareConstantArrays()
1335 const auto& rhs_array = *rhs_array_it->second; in DedupeConstantArrays() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 851 auto rhs_array = absl::make_unique<Array2D<float>>(3, 2); in TEST_P() local
895 auto rhs_array = absl::make_unique<Array2D<float>>(3, 2); in TEST_P() local
932 auto rhs_array = absl::make_unique<Array4D<float>>(2, 2, 3, 1); in TEST_P() local
976 Array3D<float> rhs_array = {{{3.f, 4.f}}}; in TEST_P() local
1033 Array4D<float> rhs_array(1, 1, 2, 2); in TEST_P() local
1255 Array4D<float> rhs_array(1, 1, 2, 2); in TEST_P() local
1319 Array4D<float> rhs_array(1, 1, 2, 2); in TEST_P() local
1385 Array4D<float> rhs_array(1, 1, 2, 3); in TEST_P() local
4642 auto rhs_array = absl::make_unique<Array2D<int8>>(3, 2); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 1184 Array2D<float> rhs_array({{5.0f, 6.0f}, {7.0f, 8.0f}}); in XLA_TEST_F() local
|