Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc82 auto* slice_b_op = new SliceOperator; in UnrollBatchMatMul3D() local
83 slice_b_op->inputs = { in UnrollBatchMatMul3D()
89 slice_b_op->outputs = {AvailableArrayName(*model, batch_name + "/slice_b")}; in UnrollBatchMatMul3D()
90 auto& slice_b_op_output = model->GetOrCreateArray(slice_b_op->outputs[0]); in UnrollBatchMatMul3D()
92 *tail_it = model->operators.emplace(*tail_it, slice_b_op) + 1; in UnrollBatchMatMul3D()
97 slice_b_op->outputs[0], in UnrollBatchMatMul3D()