HomeSort by relevance Sort by last modified time
    Searched defs:IrEmitter (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.h 53 // There are two concrete subclasses of IrEmitter: IrEmitterNested and
66 // not a subclass of gpu::IrEmitter, and in fact is better understood as an IR
68 class IrEmitter : public DfsHloVisitorWithDefault,
69 public IrBuilderMixin<IrEmitter> {
74 IrEmitter(const IrEmitter&) = delete;
75 IrEmitter& operator=(const IrEmitter&) = delete;
110 // Constructs an IrEmitter with the given IrEmitter context
    [all...]
ir_emitter.cc 57 IrEmitter::IrEmitter(const HloModuleConfig& hlo_module_config,
68 Status IrEmitter::DefaultAction(HloInstruction* hlo) {
81 Status IrEmitter::HandleConstant(HloInstruction* constant) {
85 Status IrEmitter::HandleBitcast(HloInstruction* bitcast) {
97 Status IrEmitter::HandleAddDependency(HloInstruction* add_dependency) {
109 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) {
122 Status IrEmitter::HandleSend(HloInstruction*) {
126 Status IrEmitter::HandleSendDone(HloInstruction*) {
130 Status IrEmitter::HandleRecv(HloInstruction*)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.h 59 class IrEmitter : public DfsHloVisitorWithDefault,
60 public IrBuilderMixin<IrEmitter> {
76 IrEmitter(const HloModule& hlo_module, const BufferAssignment& assignment,
84 ~IrEmitter() override;
588 TF_DISALLOW_COPY_AND_ASSIGN(IrEmitter);
ir_emitter.cc 83 IrEmitter::IrEmitter(
109 StatusOr<llvm::Function*> IrEmitter::EmitComputation(
152 void IrEmitter::InitializeIrFunction(const string& function_name) {
165 IrEmitter::~IrEmitter() {}
167 Status IrEmitter::HandleBitcast(HloInstruction* bitcast) {
175 llvm::Constant* IrEmitter::EmitGlobalForLiteral(const Literal& literal) {
191 Status IrEmitter::EmitConstantGlobals() {
214 Status IrEmitter::HandleConstant(HloInstruction* constant)
    [all...]

Completed in 689 milliseconds