Searched defs:shape_dim (Results 1 – 5 of 5) sorted by relevance
354 const int64_t shape_dim = shape.dimensions(i); in TileOffsetForDevice() local375 const int64_t shape_dim = shape.dimensions(i); in TileLimitForDevice() local673 const int64_t shape_dim = shape.dimensions(i); in TileShape() local
862 int64_t shape_dim = window_dim_to_shape_dim(dim_index); in RewriteInputWithDynamicPadding() local
92 auto shape_dim = shape->add_dim(); in BuildNode() local
784 DimensionHandle shape_dim = Dim(tensor_shape, 0); in InternalMakeShapeFromTensor() local
1987 shape_dim): argument