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