Searched defs:lhs_array (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 207 const llvm_ir::IrArray& lhs_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()
1032 const llvm_ir::IrArray& lhs_array, in EmitDotOperation()
|
| D | ir_emitter.cc | 929 llvm_ir::IrArray lhs_array(GetIrArrayFor(lhs)); in HandleDot() local
2185 llvm_ir::IrArray lhs_array( in HandleFusion() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | binop_scaling_test.cc | 115 Array4D<int> lhs_array({ in TEST_F() local
|
| D | dot_operation_test.cc | 1141 Array2D<float> lhs_array({{1.0f, 2.0f}, {3.0f, 4.0f}}); in XLA_TEST_F() 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()
774 bool HaveSameQuantizationParams(const Array& lhs_array, in HaveSameQuantizationParams()
789 bool CompareConstantArrays(const Array& lhs_array, const Array& rhs_array) { in CompareConstantArrays()
1279 const auto& lhs_array = *lhs_array_it->second; in DedupeConstantArrays() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_test.cc | 722 auto lhs_array = absl::make_unique<Array2D<float>>(4, 1); in TEST_P() local
806 auto lhs_array = absl::make_unique<Array2D<float>>(4, 3); in TEST_P() local
849 auto lhs_array = absl::make_unique<Array4D<float>>(2, 2, 3, 1); in TEST_P() local
894 Array3D<float> lhs_array = {{{1, 2, 3}}}; in TEST_P() local
943 Array4D<float> lhs_array(1, 1, 4, 4); in TEST_P() local
1165 Array4D<float> lhs_array(1, 1, 4, 4); in TEST_P() local
1229 Array4D<float> lhs_array(1, 1, 4, 4); in TEST_P() local
1295 Array4D<float> lhs_array(1, 1, 4, 4); in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.cc | 471 const llvm_ir::IrArray& lhs_array = GetIrArray(*lhs_instruction, *dot); in HandleDot() local
|