Home
last modified time | relevance | path

Searched refs:a_input_str (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc274 absl::string_view a_input_str = input_split[0]; in IsEinsumCorrectlyFormed() local
278 if (absl::StrContains(a_input_str, "...") || in IsEinsumCorrectlyFormed()
297 if (a_input_str.size() != static_cast<size_t>(a_input_shape.dim_size()) || in IsEinsumCorrectlyFormed()
306 if (CheckRepeatedDimensions(a_input_str) || in IsEinsumCorrectlyFormed()
1147 absl::string_view a_input_str = input_split[0]; in GenerateBatchMatmulContextFromEinsum() local
1191 for (int i_idx = 0, a_input_str_size = a_input_str.size(); in GenerateBatchMatmulContextFromEinsum()
1193 if (b_input_str.find(a_input_str[i_idx]) == std::string::npos) { in GenerateBatchMatmulContextFromEinsum()
1194 if (rhs_str.find(a_input_str[i_idx]) == std::string::npos) { in GenerateBatchMatmulContextFromEinsum()
1201 } else if (rhs_str.find(a_input_str[i_idx]) == std::string::npos) { in GenerateBatchMatmulContextFromEinsum()
1214 if (a_input_str.find(b_input_str[i_idx]) == std::string::npos) { in GenerateBatchMatmulContextFromEinsum()