Searched refs:input_rhs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_tensorflow_matmul.cc | 87 string input_rhs = matmul_op->inputs[1]; in Run() local
92 auto* transpose_op = FindTransposeOpWithInput(*model, input_rhs); in Run()
97 LogName(*matmul_op), input_rhs); in Run()
101 input_rhs, in Run()
103 model, AvailableArrayName(*model, input_rhs + "/transpose/perm"), in Run()
106 AvailableArrayName(*model, input_rhs + "/transpose")}; in Run()
110 DCHECK_EQ(transpose_op, FindTransposeOpWithInput(*model, input_rhs)); in Run()
115 LogName(*matmul_op), input_rhs); in Run()
118 input_rhs = transpose_op->outputs[0]; in Run()
200 fc_op->inputs = {input_lhs, input_rhs}; in Run()
|
| D | unroll_batch_matmul.cc | 33 const string& input_lhs, const string& input_rhs, in UnrollBatchMatMul3D() argument
40 const auto& input_array_b = model->GetArray(input_rhs); in UnrollBatchMatMul3D()
84 input_rhs, in UnrollBatchMatMul3D()
121 const string& input_lhs, const string& input_rhs, in UnrollBatchMatMulRecursion() argument
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()
227 string input_rhs = batch_op->inputs[1]; in Run() local
229 const auto& input_rhs_array = model->GetArray(input_rhs); in Run()
243 TransposeOperator* transpose_op = TransposeInput(input_rhs, model); in Run()
245 input_rhs = transpose_op->outputs[0]; in Run()
[all …]
|