OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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