Searched refs:PrependZerosInMajorDims (Results 1 – 2 of 2) sorted by relevance
33 xla::XlaOp PrependZerosInMajorDims(xla::XlaOp x,
115 StatusOr<std::vector<XlaOp>> PrependZerosInMajorDims( in PrependZerosInMajorDims() function143 TF_ASSIGN_OR_RETURN(auto padded_starts, PrependZerosInMajorDims(x, starts)); in DynamicSliceInMinorDims()153 TF_ASSIGN_OR_RETURN(auto padded_starts, PrependZerosInMajorDims(x, starts)); in DynamicUpdateSliceInMinorDims()