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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
thunk.h 23 #include "tensorflow/compiler/xla/service/hlo_instruction.h"
59 // The hlo_instruction argument is meant to be the instruction this thunk was
61 // to Thunk::hlo_instruction, so it can be null.
62 explicit Thunk(Kind kind, const HloInstruction* hlo_instruction)
63 : kind_(kind), hlo_instruction_(hlo_instruction) {}
69 const HloInstruction* hlo_instruction() const { return hlo_instruction_; } function in class:xla::gpu::Thunk
  /external/tensorflow/tensorflow/compiler/xla/service/
user_computation.cc 29 #include "tensorflow/compiler/xla/service/hlo_instruction.h"
2882 HloInstruction* hlo_instruction; local
    [all...]

Completed in 106 milliseconds