Searched refs:input_lhs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 33 const string& input_lhs, const string& input_rhs, in UnrollBatchMatMul3D() argument
39 const auto& input_array_a = model->GetArray(input_lhs); in UnrollBatchMatMul3D()
52 input_lhs, in UnrollBatchMatMul3D()
121 const string& input_lhs, const string& input_rhs, in UnrollBatchMatMulRecursion() argument
125 const auto& input_array_a = model->GetArray(input_lhs); in UnrollBatchMatMulRecursion()
135 UnrollBatchMatMul3D(input_lhs, input_rhs, batch_op, new_batch_prefix, in UnrollBatchMatMulRecursion()
144 input_lhs, input_rhs, batch_op, model, tail_it, new_batch_prefix); in UnrollBatchMatMulRecursion()
226 string input_lhs = batch_op->inputs[0]; in Run() local
228 const auto& input_lhs_array = model->GetArray(input_lhs); in Run()
235 TransposeOperator* transpose_op = TransposeInput(input_lhs, model); in Run()
[all …]
|
| D | resolve_tensorflow_matmul.cc | 86 string input_lhs = matmul_op->inputs[0]; in Run() local
175 input_lhs = previous_op->inputs[0]; in Run()
200 fc_op->inputs = {input_lhs, input_rhs}; in Run()
|