Searched defs:x_rank (Results 1 – 2 of 2) sorted by relevance
411 const int64_t x_rank = x_config.size(); in Einsum() local595 absl::string_view einsum_config, int64_t x_rank, int64_t y_rank) { in ParseEinsumString()
2061 def _flatten_with_inner_dim(x, dim, x_rank): argument2075 def _unflatten_with_inner_dim(x, dim, x_rank, stack_size): argument