OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:GetIrArray
(Results
1 - 6
of
6
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.h
108
// Helper for calling HloToIrBindings::
GetIrArray
.
113
//
GetIrArray
(hlo, hlo).
114
llvm_ir::IrArray
GetIrArray
(const HloInstruction& inst,
117
return bindings_.
GetIrArray
(inst, consumer, shape_index);
hlo_to_ir_bindings.h
86
//
GetIrArray
(hlo, hlo).
87
llvm_ir::IrArray
GetIrArray
(const HloInstruction& hlo,
ir_emitter.cc
73
return
GetIrArray
(*operand, *hlo)
151
llvm_ir::EmitTuple(
GetIrArray
(*tuple, *tuple), base_ptrs, &ir_builder_,
428
llvm_ir::EmitTupleSelect(
GetIrArray
(*select, *select),
429
GetIrArray
(*pred, *select),
444
const llvm_ir::IrArray& target_array =
GetIrArray
(*dot, *dot);
445
const llvm_ir::IrArray& lhs_array =
GetIrArray
(*lhs_instruction, *dot);
446
const llvm_ir::IrArray& rhs_array =
GetIrArray
(*rhs_instruction, *dot);
673
GetIrArray
(*arg, *reduce)
690
parameter_arrays.push_back(
GetIrArray
(*operand, *fusion));
727
return
GetIrArray
(*operand, *random
[
all
...]
ir_emitter_nested.cc
119
return llvm_ir::LoopEmitter(element_generator,
GetIrArray
(hlo, hlo),
ir_emitter_unnested.cc
511
parameter_arrays.push_back(
GetIrArray
(*operand, *fusion));
573
operand_arrays.push_back(
GetIrArray
(*operand, *fusion));
584
llvm_ir::IrArray output_array =
GetIrArray
(*fusion, *fusion);
[
all
...]
hlo_to_ir_bindings.cc
196
llvm_ir::IrArray HloToIrBindings::
GetIrArray
(const HloInstruction& hlo,
Completed in 84 milliseconds