Searched refs:slice_dimensions (Results 1 – 3 of 3) sorted by relevance
1529 std::vector<int64> slice_dimensions(rank, 1); in HandleSort() local1530 slice_dimensions[sort_dim] = sort_dim_elements; in HandleSort()1537 sorted_literal.Reshape(slice_dimensions)); in HandleSort()1540 slice_dimensions)); in HandleSort()
290 for (const HloInstructionProto::SliceDimensions& slice_dimensions : in CreateFromProto() local291 proto.slice_dimensions()) { in CreateFromProto()292 slice_starts.push_back(slice_dimensions.start()); in CreateFromProto()293 slice_limits.push_back(slice_dimensions.limit()); in CreateFromProto()294 slice_strides.push_back(slice_dimensions.stride()); in CreateFromProto()
94 repeated SliceDimensions slice_dimensions = 17; field