Searched defs:unnested_hlo (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.h | 236 HloInstruction* unnested_hlo) { in GetGeneratorForOperandIrArrays()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 296 const HloInstruction* unnested_hlo = hlo; in GetIndexTypeForKernel() local
707 const HloInstruction* unnested_hlo, const IrArray::Index& index, in EmitExtraOutputsForReduce()
2643 HloInstruction* unnested_hlo, HloInstruction* reduce_inst, int reduce_idx, in EmitPrologueForOneReduction()
2698 HloInstruction* unnested_hlo, KernelCodegenInfo* kernel_info) { in EmitPrologueForReduction()
2779 HloInstruction* unnested_hlo, KernelCodegenInfo* kernel_info) { in EmitEpilogueForReduction()
2853 HloInstruction* unnested_hlo, const llvm_ir::IrArray::Index& index, in EmitTileElementForReduction()
3051 HloInstruction* unnested_hlo, absl::Span<const int64> tiled_param_ids, in EmitKernel()
3570 const HloInstruction* unnested_hlo, const Shape& op_shape, in NumInputsInvolveInOnlyElementwiseOps()
3582 int64 NumInputsWithMoreElementsThan(const HloInstruction* unnested_hlo, in NumInputsWithMoreElementsThan()
3597 bool IsUnrollingColumnReductionBeneficial(const HloInstruction* unnested_hlo, in IsUnrollingColumnReductionBeneficial()
[all …]
|