HomeSort by relevance Sort by last modified time
    Searched refs:IsReductionToVector (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_fusible.cc 67 if (IsReductionToVector(*operand)) {
76 IsReductionToVector(*instr.fused_expression_root())) {
86 return IsReduceInputFusion(instr) || IsReductionToVector(instr);
101 if (IsReductionToVector(*inst)) {
117 if (IsReductionToVector(*element_instr)) {
130 if (IsReductionToVector(*instr_1) && IsReductionToVector(*instr_2) &&
175 !IsReductionToVector(instr)) ||
ir_emission_utils.h 151 bool IsReductionToVector(const HloInstruction& reduce);
instruction_fusion.cc 205 if (IsReductionToVector(*producer)) {
281 if (IsReductionToVector(*consumer) ||
ir_emission_utils.cc 160 bool IsReductionToVector(const HloInstruction& reduce) {
multi_output_fusion.cc 95 (IsReductionToVector(*instr2) &&
ir_emitter_unnested.cc 727 if (IsReductionToVector(*reduce)) {
    [all...]

Completed in 1540 milliseconds