Searched defs:new_dim (Results 1 – 12 of 12) sorted by relevance
45 shape_inference::DimensionHandle new_dim; in __anon141003b80102() local
29 shape_inference::DimensionHandle new_dim; in _UpdatePartitionDim() local
121 unsigned new_dim = affine_map.operands.size(); in TileLoop() local
197 const int new_dim = in ToRuntimeDims() local
248 auto* new_dim = new_backward_conv_window.mutable_dimensions(i); in CanonicalizeBackwardFilterConvolution() local
544 int64_t new_dim = orig_dim + low + high; in __anonb720852f0302() local
537 auto new_dim = new_window.add_dimensions(); in HandleConvolution() local
1979 int64_t new_dim = lhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local1985 int64_t new_dim = lhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local1992 int64_t new_dim = rhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local1998 int64_t new_dim = rhs_dimension_map[dim]; in RemoveDegenerateDimensionFromDot() local6463 auto new_dim = swapped_window.add_dimensions(); in SwapConvOperands() local
661 DimensionHandle new_dim, ShapeHandle* out) { in ReplaceDim()
235 DimensionHandle new_dim = c->RankKnown(input_shape) in EinsumShape() local
1195 DimensionHandle new_dim = GetUnknownOutputDim(node, port_index, d); in OutputAsUnion() local
192 WindowDimension* new_dim = new_window.add_dimensions(); in GenNewWindow() local