Searched refs:reduced_dims (Results 1 – 3 of 3) sorted by relevance
60 OutputDims* output_dims, ReducedDims* reduced_dims) {66 (*reduced_dims)[reduceIndex] = input_dims[i];79 Sizes<>*, array<Index, Rank>* reduced_dims) {82 (*reduced_dims)[i] = input_dims[i];
2992 absl::Span<const int64> reduced_dims = in EmitBlock() local3008 reduced_dims[i] - (dims_in_tile[i] - 1) * tile_size_for_dim; in EmitBlock()3107 absl::Span<const int64> reduced_dims = in EmitKernel() local3110 *unnested_hlo, param_arrays, param_shmem_buffers, reduced_dims, in EmitKernel()
1367 absl::flat_hash_set<int64> reduced_dims(reduce.dimensions().begin(), in ReductionPreservesLayout() local1375 if (reduced_dims.contains(i)) { in ReductionPreservesLayout()1388 if (!reduced_dims.contains(operand_dim)) { in ReductionPreservesLayout()