Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc50 auto* slice_a_op = new SliceOperator; in UnrollBatchMatMul3D() local
51 slice_a_op->inputs = { in UnrollBatchMatMul3D()
57 slice_a_op->outputs = {AvailableArrayName(*model, batch_name + "/slice_a")}; in UnrollBatchMatMul3D()
58 auto& slice_a_op_output = model->GetOrCreateArray(slice_a_op->outputs[0]); in UnrollBatchMatMul3D()
60 *tail_it = model->operators.emplace(*tail_it, slice_a_op) + 1; in UnrollBatchMatMul3D()
65 slice_a_op->outputs[0], in UnrollBatchMatMul3D()