Searched refs:padded_starts (Results 1 – 1 of 1) sorted by relevance
121 std::vector<XlaOp> padded_starts(n_dims, zero); in PrependZerosInMajorDims() local123 padded_starts[n_dims - starts.size() + i] = starts[i]; in PrependZerosInMajorDims()125 return padded_starts; in PrependZerosInMajorDims()143 TF_ASSIGN_OR_RETURN(auto padded_starts, PrependZerosInMajorDims(x, starts)); in DynamicSliceInMinorDims()145 return DynamicSlice(x, padded_starts, padded_sizes); in DynamicSliceInMinorDims()153 TF_ASSIGN_OR_RETURN(auto padded_starts, PrependZerosInMajorDims(x, starts)); in DynamicUpdateSliceInMinorDims()154 return DynamicUpdateSlice(x, update, padded_starts); in DynamicUpdateSliceInMinorDims()