Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc257 std::vector<std::string> equation_split = absl::StrSplit(equation, "->"); in IsEinsumCorrectlyFormed() local
259 if (equation_split.empty()) { in IsEinsumCorrectlyFormed()
264 absl::StrSplit(equation_split[0], ','); in IsEinsumCorrectlyFormed()
267 if (op_info.inputs_size() != 2 || equation_split.size() != 2) { in IsEinsumCorrectlyFormed()
273 absl::string_view rhs_str = equation_split[1]; in IsEinsumCorrectlyFormed()
1140 std::vector<std::string> equation_split = in GenerateBatchMatmulContextFromEinsum() local
1143 absl::StrSplit(equation_split[0], ','); in GenerateBatchMatmulContextFromEinsum()
1146 absl::string_view rhs_str = equation_split[1]; in GenerateBatchMatmulContextFromEinsum()