Searched refs:y_rank (Results 1 – 3 of 3) sorted by relevance
1034 y_rank = array_ops.rank(y)1036 y_rank - event_ndims : y_rank - min_event_ndims]
119 absl::string_view einsum_config, int64_t x_rank, int64_t y_rank);
412 const int64_t y_rank = y_config.size(); in Einsum() local477 for (int64_t i = 0; i < y_rank; ++i) { in Einsum()595 absl::string_view einsum_config, int64_t x_rank, int64_t y_rank) { in ParseEinsumString() argument669 /*is_input_config=*/true, y_rank, in ParseEinsumString()