Searched refs:einsum_config_numeric (Results 1 – 1 of 1) sorted by relevance
284 std::array<std::vector<int64>, 3> einsum_config_numeric; in ParseEinsumString() local302 auto& x_config = einsum_config_numeric[0]; in ParseEinsumString()313 auto& y_config = einsum_config_numeric[1]; in ParseEinsumString()319 auto& output_config = einsum_config_numeric[2]; in ParseEinsumString()325 return einsum_config_numeric; in ParseEinsumString()332 TF_ASSIGN_OR_RETURN(auto einsum_config_numeric, in Einsum()334 return Einsum(x, einsum_config_numeric[0], y, einsum_config_numeric[1], in Einsum()335 einsum_config_numeric[2], precision); in Einsum()