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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.h 108 // Helper for calling HloToIrBindings::GetIrArray.
113 // GetIrArray(hlo, hlo).
114 llvm_ir::IrArray GetIrArray(const HloInstruction& inst,
117 return bindings_.GetIrArray(inst, consumer, shape_index);
hlo_to_ir_bindings.h 86 // GetIrArray(hlo, hlo).
87 llvm_ir::IrArray GetIrArray(const HloInstruction& hlo,
ir_emitter.cc 73 return GetIrArray(*operand, *hlo)
151 llvm_ir::EmitTuple(GetIrArray(*tuple, *tuple), base_ptrs, &ir_builder_,
428 llvm_ir::EmitTupleSelect(GetIrArray(*select, *select),
429 GetIrArray(*pred, *select),
444 const llvm_ir::IrArray& target_array = GetIrArray(*dot, *dot);
445 const llvm_ir::IrArray& lhs_array = GetIrArray(*lhs_instruction, *dot);
446 const llvm_ir::IrArray& rhs_array = GetIrArray(*rhs_instruction, *dot);
673 GetIrArray(*arg, *reduce)
690 parameter_arrays.push_back(GetIrArray(*operand, *fusion));
727 return GetIrArray(*operand, *random
    [all...]
ir_emitter_nested.cc 119 return llvm_ir::LoopEmitter(element_generator, GetIrArray(hlo, hlo),
ir_emitter_unnested.cc 511 parameter_arrays.push_back(GetIrArray(*operand, *fusion));
573 operand_arrays.push_back(GetIrArray(*operand, *fusion));
584 llvm_ir::IrArray output_array = GetIrArray(*fusion, *fusion);
    [all...]
hlo_to_ir_bindings.cc 196 llvm_ir::IrArray HloToIrBindings::GetIrArray(const HloInstruction& hlo,

Completed in 84 milliseconds