Searched defs:shape_dim (Results 1 – 4 of 4) sorted by relevance
174 const int64 shape_dim = shape.dimensions(i); in TileOffsetForDevice() local193 const int64 shape_dim = shape.dimensions(i); in TileLimitForDevice() local
90 auto shape_dim = shape->add_dim(); in BuildNode() local
833 DimensionHandle shape_dim = Dim(tensor_shape, 0); in InternalMakeShapeFromTensor() local
1569 shape_dim): argument