Searched refs:slice_size_a (Results 1 – 1 of 1) sorted by relevance
46 std::vector<int> slice_size_a = input_array_a.shape().dims(); in UnrollBatchMatMul3D() local48 slice_size_a[i] = 1; in UnrollBatchMatMul3D()55 CreateInt32Array(model, batch_name + "/slice_a/slice/size", slice_size_a), in UnrollBatchMatMul3D()