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
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