Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc5247 int64 can_be_vectorized = 0; in IsUnrollingColumnReductionBeneficial() local
5265 can_be_vectorized++; in IsUnrollingColumnReductionBeneficial()
5272 can_be_vectorized += NumInputsInvolveInOnlyElementwiseOps(fusion, input_shape, in IsUnrollingColumnReductionBeneficial()
5280 return can_be_vectorized >= cannot_be_vectorized; in IsUnrollingColumnReductionBeneficial()