Searched refs:slice_size_b (Results 1 – 1 of 1) sorted by relevance
78 std::vector<int> slice_size_b = input_array_b.shape().dims(); in UnrollBatchMatMul3D() local80 slice_size_b[i] = 1; in UnrollBatchMatMul3D()87 CreateInt32Array(model, batch_name + "/slice_b/slice/size", slice_size_b), in UnrollBatchMatMul3D()