Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2524 bool is_row_reduction) in ReductionCodegenInfo() argument
2528 is_row_reduction_(is_row_reduction) {} in ReductionCodegenInfo()
3654 bool is_row_reduction = true; in ComputeMappingSchemeAndReductionKind() local
3679 is_row_reduction = false; in ComputeMappingSchemeAndReductionKind()
3717 VLOG(10) << "is_row_reduction " << is_row_reduction << depth << " " << height in ComputeMappingSchemeAndReductionKind()
3726 return std::make_tuple(mapping_scheme, is_row_reduction); in ComputeMappingSchemeAndReductionKind()
3771 bool is_row_reduction; in EmitReductionToVector() local
3773 std::tie(mapping_scheme, is_row_reduction) = in EmitReductionToVector()
3775 ReductionCodegenInfo reduction_info(&mapping_scheme, is_row_reduction); in EmitReductionToVector()