Searched refs:num_reduced_major (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 3530 int num_reduced_major = 1, num_kept = 1, num_reduced_minor = 1; in GetReductionToVectorDimensions() local
3532 return std::make_tuple(num_reduced_major, num_kept, num_reduced_minor); in GetReductionToVectorDimensions()
3543 num_reduced_major *= curr_dim_size; in GetReductionToVectorDimensions()
3553 return std::make_tuple(num_reduced_major, num_kept, num_reduced_minor); in GetReductionToVectorDimensions()
3663 int64 num_reduced_major, num_kept, num_reduced_minor; in ComputeMappingSchemeAndReductionKind() local
3664 std::tie(num_reduced_major, num_kept, num_reduced_minor) = in ComputeMappingSchemeAndReductionKind()
3677 height = num_reduced_major; in ComputeMappingSchemeAndReductionKind()
3702 depth = num_reduced_major; in ComputeMappingSchemeAndReductionKind()
|