Home
last modified time | relevance | path

Searched defs:lhs_array (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_tensorflow_matmul.cc83 Array& lhs_array = model->GetArray(input_lhs); in Run() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc237 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()
1232 const llvm_ir::IrArray& lhs_array, in EmitDotOperation()
Dir_emitter.cc849 llvm_ir::IrArray lhs_array(GetIrArrayFor(lhs)); in HandleDot() local
2100 llvm_ir::IrArray lhs_array( in HandleFusion() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbinop_scaling_test.cc115 Array4D<int> lhs_array({ in TEST_F() local
Ddot_operation_test.cc1181 Array2D<float> lhs_array({{1.0f, 2.0f}, {3.0f, 4.0f}}); in XLA_TEST_F() local
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc774 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()
802 bool HaveSameQuantizationParams(const Array& lhs_array, in HaveSameQuantizationParams()
817 bool CompareConstantArrays(const Array& lhs_array, const Array& rhs_array) { in CompareConstantArrays()
1310 const auto& lhs_array = *lhs_array_it->second; in DedupeConstantArrays() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc799 auto lhs_array = absl::make_unique<Array2D<float>>(4, 1); in TEST_P() local
883 auto lhs_array = absl::make_unique<Array2D<float>>(4, 3); in TEST_P() local
926 auto lhs_array = absl::make_unique<Array4D<float>>(2, 2, 3, 1); in TEST_P() local
971 Array3D<float> lhs_array = {{{1, 2, 3}}}; in TEST_P() local
1020 Array4D<float> lhs_array(1, 1, 4, 4); in TEST_P() local
1242 Array4D<float> lhs_array(1, 1, 4, 4); in TEST_P() local
1306 Array4D<float> lhs_array(1, 1, 4, 4); in TEST_P() local
1372 Array4D<float> lhs_array(1, 1, 4, 4); in TEST_P() local
4632 auto lhs_array = absl::make_unique<Array2D<int16>>(4, 3); in TEST_F() local