OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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