Searched refs:batch_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 34 const BatchMatMulOperator* batch_op, const std::vector<int> batch, in UnrollBatchMatMul3D() argument
38 absl::StrCat(batch_op->outputs[0], "_b", absl::StrJoin(batch, "-")); in UnrollBatchMatMul3D()
122 const BatchMatMulOperator* batch_op, Model* model, 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()
149 batch_op->outputs[0], "_b", absl::StrJoin(new_batch_prefix, "-")); in UnrollBatchMatMulRecursion()
221 const auto* batch_op = in Run() local
226 string input_lhs = batch_op->inputs[0]; in Run()
227 string input_rhs = batch_op->inputs[1]; in Run()
234 if (batch_op->adj_x) { in Run()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization_test.cc | 164 const string& batch_op, const string& map_input_name) { in CheckNotVectorized() argument
166 ASSERT_EQ(graph_utils::FindAllGraphNodesWithOp(batch_op, output).size(), 1); in CheckNotVectorized()
170 output.node(graph_utils::FindGraphNodeWithOp(batch_op, output)); in CheckNotVectorized()
|