Home
last modified time | relevance | path

Searched refs:reduction_output_shape_indices (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2654 InlinedVector<ShapeIndex, 1>* reduction_output_shape_indices = in EmitPrologueForOneReduction() local
2656 reduction_output_shape_indices->push_back(std::move(output_shape_index)); in EmitPrologueForOneReduction()
2787 absl::Span<const ShapeIndex> reduction_output_shape_indices = in EmitEpilogueForReduction() local
2822 reduction_output_shape_indices[i]), in EmitEpilogueForReduction()
2826 reduction_output_shape_indices[i]) in EmitEpilogueForReduction()