Searched defs:shape_dim (Results 1 – 5 of 5) sorted by relevance
278 const int64 shape_dim = shape.dimensions(i); in TileOffsetForDevice() local299 const int64 shape_dim = shape.dimensions(i); in TileLimitForDevice() local587 const int64 shape_dim = shape.dimensions(i); in TileShape() local
764 int64 shape_dim = window_dim_to_shape_dim(dim_index); in RewriteInputWithDynamicPadding() local
92 auto shape_dim = shape->add_dim(); in BuildNode() local
777 DimensionHandle shape_dim = Dim(tensor_shape, 0); in InternalMakeShapeFromTensor() local
1968 shape_dim): argument