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 52 // There are two concrete subclasses of IrEmitter: IrEmitterNested and
65 // not a subclass of gpu::IrEmitter, and in fact is better understood as an IR
67 class IrEmitter : public DfsHloVisitorWithDefault {
69 IrEmitter(const IrEmitter&) = delete;
70 IrEmitter& operator=(const IrEmitter&) = delete;
102 // Constructs an IrEmitter with the given IrEmitter context.
103 // ir_emitter_context is owned by the caller and should outlive the IrEmitter
    [all...]
ir_emitter.cc 55 IrEmitter::IrEmitter(const HloModuleConfig& hlo_module_config,
69 Status IrEmitter::DefaultAction(HloInstruction* hlo) {
83 Status IrEmitter::HandleConstant(HloInstruction* constant) {
100 Status IrEmitter::HandleBitcast(HloInstruction* bitcast) {
112 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) {
125 Status IrEmitter::HandleSort(HloInstruction*) {
130 Status IrEmitter::HandleSend(HloInstruction*) {
134 Status IrEmitter::HandleSendDone(HloInstruction*) {
138 Status IrEmitter::HandleRecv(HloInstruction*)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.h 58 class IrEmitter : public DfsHloVisitorWithDefault {
73 IrEmitter(const HloModule& hlo_module, const BufferAssignment& assignment,
81 ~IrEmitter() override;
544 TF_DISALLOW_COPY_AND_ASSIGN(IrEmitter);
ir_emitter.cc 81 IrEmitter::IrEmitter(
106 StatusOr<llvm::Function*> IrEmitter::EmitComputation(
139 void IrEmitter::InitializeIrFunction(const string& function_name) {
154 IrEmitter::~IrEmitter() {}
156 Status IrEmitter::HandleBitcast(HloInstruction* bitcast) {
165 Status IrEmitter::HandleConstant(HloInstruction* constant) {
208 Status IrEmitter::HandleCopy(HloInstruction* copy) {
220 int IrEmitter::MinimumAlignmentForBufferSize(int64 buffer_size)
    [all...]

Completed in 98 milliseconds