Home
last modified time | relevance | path

Searched refs:slice_b_reshape_op (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc95 auto* slice_b_reshape_op = new TensorFlowReshapeOperator; in UnrollBatchMatMul3D() local
96 slice_b_reshape_op->inputs = { in UnrollBatchMatMul3D()
100 slice_b_reshape_op->outputs = { in UnrollBatchMatMul3D()
103 model->GetOrCreateArray(slice_b_reshape_op->outputs[0]); in UnrollBatchMatMul3D()
105 *tail_it = model->operators.emplace(*tail_it, slice_b_reshape_op) + 1; in UnrollBatchMatMul3D()
110 slice_b_reshape_op->outputs[0]}; in UnrollBatchMatMul3D()