Searched defs:output_shape_size (Results 1 – 2 of 2) sorted by relevance
411 int64 output_shape_size) { in SelectElementsBasedOnId()490 const int64 output_shape_size = output_shape.size(); in LowerAllGatherOp() local
440 int64_t rhs_shape_size, int64_t lhs_shape_size, int64_t output_shape_size, in GetWindowedEinsumConfiguration()