Searched defs:original_dim (Results 1 – 5 of 5) sorted by relevance
222 const std::optional<int64_t> original_dim, in quantile_output_shape()375 const std::optional<int64_t> original_dim, in quantile_out_impl()400 const std::optional<int64_t> original_dim, in quantile_impl()
14 std::vector<int64_t> original_dim(ndim); in infer_dense_strides_dim_last() local
104 XlaOp original_dim = GetDimensionSize(result, dim); in ReconsileBranchDifference() local
233 const int original_dim = sparsity.traversal_order()->Get(i); in VerifyAndCountElements() local
558 const int64_t original_dim = input_indices_t(i, j); in Split() local