Searched defs:new_dim (Results 1 – 13 of 13) sorted by relevance
45 shape_inference::DimensionHandle new_dim; in __anonea6454330102() local
29 shape_inference::DimensionHandle new_dim; in _UpdatePartitionDim() local
119 unsigned new_dim = affine_map.operands.size(); in TileLoop() local
468 int64 new_dim = orig_dim + low + high; in __anon3500048a0302() local
245 auto* new_dim = new_backward_conv_window.mutable_dimensions(i); in CanonicalizeBackwardFilterConvolution() local
531 auto new_dim = new_window.add_dimensions(); in HandleConvolution() local
1765 int64 new_dim = lhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local1771 int64 new_dim = lhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local1778 int64 new_dim = rhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local1784 int64 new_dim = rhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local5327 auto new_dim = swapped_window.add_dimensions(); in SwapConvOperands() local
93 int64 new_dim = to_merge->tile_assignment().dim(i); in MergeSharding() local
654 DimensionHandle new_dim, ShapeHandle* out) { in ReplaceDim()
235 DimensionHandle new_dim = c->RankKnown(input_shape) in EinsumShape() local
1158 DimensionHandle new_dim = GetUnknownOutputDim(node, port_index, d); in OutputAsUnion() local
179 WindowDimension* new_dim = new_window.add_dimensions(); in GenNewWindow() local
332 const int new_dim = in ToRuntimeDims() local