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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_unnested.h 211 const HloInstruction* unnested_hlo, const llvm_ir::IrArray::Index& index,
217 // Prerequisite: `IsReductionToVector(*unnested_hlo)`
218 Status EmitReductionToVector(HloInstruction* unnested_hlo);
221 // the reduction is a row reduction. For an un-fused reduce op, unnested_hlo
223 // unnested_hlo is the fusion instruction while first_reduce is the first
226 ComputeMappingSchemeAndReductionKind(const HloInstruction* unnested_hlo,
247 LaunchDimensions EmitKernel(HloInstruction* unnested_hlo,
270 void EmitTileElementForReduction(HloInstruction* unnested_hlo,
276 void EmitPrologueForReduction(HloInstruction* unnested_hlo,
278 void EmitPrologueForOneReduction(HloInstruction* unnested_hlo,
    [all...]
ir_emitter_unnested.cc 296 const HloInstruction* unnested_hlo = hlo; local
299 unnested_hlo = computation->FusionInstruction();
321 if (!shape_in_range(unnested_hlo->shape())) {
330 if (!absl::c_all_of(unnested_hlo->operands(), hlo_shape_in_range)) {
335 if (unnested_hlo->opcode() == HloOpcode::kFusion) {
337 unnested_hlo->fused_instructions_computation()->instructions(),
707 const HloInstruction* unnested_hlo, const IrArray::Index& index,
712 GetIrArray(*unnested_hlo, *unnested_hlo, extra_output_gens[i].second)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.h 236 HloInstruction* unnested_hlo) {
237 return [=]() { return GetIrArraysForOperandsOf(unnested_hlo); };

Completed in 4119 milliseconds