Searched defs:out_dim_sizes (Results 1 – 2 of 2) sorted by relevance
86 std::vector<int64> out_dim_sizes; in SparseTensorReduceHelper() local322 auto out_dim_sizes = reduction.reduced_shape.dim_sizes(); in Compute() local
476 const LocalOp& operand, absl::Span<const int64> out_dim_sizes, in BroadcastInDim()