Searched refs:padded_starts (Results 1 – 1 of 1) sorted by relevance
100 std::vector<XlaOp> padded_starts(n_dims, zero); in PrependZerosInMajorDims() local102 padded_starts[n_dims - starts.size() + i] = starts[i]; in PrependZerosInMajorDims()104 return padded_starts; in PrependZerosInMajorDims()122 TF_ASSIGN_OR_RETURN(auto padded_starts, PrependZerosInMajorDims(x, starts)); in DynamicSliceInMinorDims()124 return DynamicSlice(x, padded_starts, padded_sizes); in DynamicSliceInMinorDims()132 TF_ASSIGN_OR_RETURN(auto padded_starts, PrependZerosInMajorDims(x, starts)); in DynamicUpdateSliceInMinorDims()133 return DynamicUpdateSlice(x, update, padded_starts); in DynamicUpdateSliceInMinorDims()