Searched defs:shape_dim (Results 1 – 7 of 7) sorted by relevance
440 const int64_t shape_dim = shape.dimensions(i); in TileOffsetForDevice() local460 const int64_t shape_dim = shape.dimensions(i); in TileLimitForDevice() local770 const int64_t shape_dim = shape.dimensions(i); in TileShape() local
917 int64_t shape_dim = window_dim_to_shape_dim(dim_index); in RewriteInputWithDynamicPadding() local
96 auto shape_dim = shape->add_dim(); in BuildNode() local
784 DimensionHandle shape_dim = Dim(tensor_shape, 0); in InternalMakeShapeFromTensor() local
1731 std::vector<int64_t> shape_dim(original_shape.dimensions().begin(), in SplitReshapeHelper() local1754 std::vector<int64_t> shape_dim(original_shape.dimensions().begin(), in MergeReshapeHelper() local
1986 shape_dim): argument
1212 int64_t shape_dim = operand_type.getRank(); in matchAndRewrite() local