Searched refs:kMatrixRank (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 285 constexpr int kMatrixRank = 2; in IsEinsumCorrectlyFormed() local
291 a_input.shape(), std::max(kMatrixRank, a_input.shape().dim_size()), in IsEinsumCorrectlyFormed()
294 b_input.shape(), std::max(kMatrixRank, b_input.shape().dim_size()), in IsEinsumCorrectlyFormed()
1150 constexpr int kMatrixRank = 2; in GenerateBatchMatmulContextFromEinsum() local
1156 a_input.shape(), std::max(kMatrixRank, a_input.shape().dim_size()), in GenerateBatchMatmulContextFromEinsum()
1159 b_input.shape(), std::max(kMatrixRank, b_input.shape().dim_size()), in GenerateBatchMatmulContextFromEinsum()
1163 (a_input.shape().dim_size() < kMatrixRank) || in GenerateBatchMatmulContextFromEinsum()
1164 (b_input.shape().dim_size() < kMatrixRank); in GenerateBatchMatmulContextFromEinsum()
|