Searched defs:rhs_dim (Results 1 – 3 of 3) sorted by relevance
108 const int rhs_dim = extended_rhs_shape.Dims(i); in ResizeOutputTensor() local367 const int rhs_dim = extended_rhs_shape.Dims(i); in Prepare() local
31 inline int broadcast_dim(int lhs_dim, int rhs_dim) { in broadcast_dim()
415 const int64 rhs_dim = rhs_dimension_number(dim_name); in Einsum() local