Searched refs:a_input (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 271 const auto& a_input = op_info.inputs(0); in IsEinsumCorrectlyFormed() local
291 a_input.shape(), std::max(kMatrixRank, a_input.shape().dim_size()), in IsEinsumCorrectlyFormed()
1144 const auto& a_input = op_info.inputs(0); in GenerateBatchMatmulContextFromEinsum() local
1156 a_input.shape(), std::max(kMatrixRank, a_input.shape().dim_size()), in GenerateBatchMatmulContextFromEinsum()
1163 (a_input.shape().dim_size() < kMatrixRank) || in GenerateBatchMatmulContextFromEinsum()
1177 a_matrix->set_dtype(a_input.dtype()); in GenerateBatchMatmulContextFromEinsum()
1256 const auto& a_input = op_info.inputs(0); in CountBatchMatMulOperations() local
1274 a_input.shape(), std::max(matrix_rank, a_input.shape().dim_size()), in CountBatchMatMulOperations()
1281 (a_input.shape().dim_size() < matrix_rank) || in CountBatchMatMulOperations()
1328 a_matrix->set_dtype(a_input.dtype()); in CountBatchMatMulOperations()
|
| /external/llvm-project/mlir/docs/ |
| D | DeclarativeRewrites.md | 119 AnyType:$a_input,
133 succeeds, we bind the `$input` symbol to the op's only input (`$a_input`) and
|