Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc63 auto* slice_a_reshape_op = new TensorFlowReshapeOperator; in UnrollBatchMatMul3D() local
64 slice_a_reshape_op->inputs = { in UnrollBatchMatMul3D()
68 slice_a_reshape_op->outputs = { in UnrollBatchMatMul3D()
71 model->GetOrCreateArray(slice_a_reshape_op->outputs[0]); in UnrollBatchMatMul3D()
73 *tail_it = model->operators.emplace(*tail_it, slice_a_reshape_op) + 1; in UnrollBatchMatMul3D()
109 matmul_op->inputs = {slice_a_reshape_op->outputs[0], in UnrollBatchMatMul3D()