Searched refs:y_rank (Results 1 – 3 of 3) sorted by relevance
108 absl::string_view einsum_config, int64 x_rank, int64 y_rank);
379 const int64 y_rank = y_config.size(); in Einsum() local444 for (int64 i = 0; i < y_rank; ++i) { in Einsum()559 absl::string_view einsum_config, int64 x_rank, int64 y_rank) { in ParseEinsumString() argument633 /*is_input_config=*/true, y_rank, in ParseEinsumString()
1040 y_rank = array_ops.rank(y)1042 y_rank - event_ndims : y_rank - min_event_ndims]