Searched refs:batch_name (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 37 const std::string batch_name = in UnrollBatchMatMul3D() local
53 CreateInt32Array(model, batch_name + "/slice_a/slice/begin", in UnrollBatchMatMul3D()
55 CreateInt32Array(model, batch_name + "/slice_a/slice/size", slice_size_a), in UnrollBatchMatMul3D()
57 slice_a_op->outputs = {AvailableArrayName(*model, batch_name + "/slice_a")}; in UnrollBatchMatMul3D()
66 CreateInt32Array(model, batch_name + "/slice_a/reshape/shape", in UnrollBatchMatMul3D()
69 AvailableArrayName(*model, batch_name + "/slice_a/reshape")}; in UnrollBatchMatMul3D()
85 CreateInt32Array(model, batch_name + "/slice_b/slice/begin", in UnrollBatchMatMul3D()
87 CreateInt32Array(model, batch_name + "/slice_b/slice/size", slice_size_b), in UnrollBatchMatMul3D()
89 slice_b_op->outputs = {AvailableArrayName(*model, batch_name + "/slice_b")}; in UnrollBatchMatMul3D()
98 CreateInt32Array(model, batch_name + "/slice_b/reshape/shape", in UnrollBatchMatMul3D()
[all …]
|