| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| llvm_util.h | 90 string IrName(string a);
91 string IrName(tensorflow::StringPiece a, tensorflow::StringPiece b);
92 string IrName(const HloInstruction* a, tensorflow::StringPiece b = "");
|
| llvm_loop.cc | 169 return llvm_ir::IrName(prefix_, llvm_ir::IrName(name, suffix_));
254 llvm_ir::IrName(suffix, tensorflow::strings::StrCat(dimension)));
|
| llvm_util.cc | 531 string IrName(string a) {
536 string IrName(tensorflow::StringPiece a, tensorflow::StringPiece b) {
538 return IrName(tensorflow::strings::StrCat(a, ".", b));
540 return IrName(tensorflow::strings::StrCat(a, b));
543 string IrName(const HloInstruction* a, tensorflow::StringPiece b) {
544 return IrName(a->name(), b);
|
| ops.cc | 156 fusion_output_array, launch_dimensions, IrName(fusion), ir_builder);
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| elemental_ir_emitter.cc | 123 llvm_ir::IrName(hlo));
|
| ir_emitter.cc | 74 using llvm_ir::IrName;
161 AsStringRef(IrName(bitcast)));
558 llvm_ir::ForLoopNest loops(IrName(reduce_window, "inner"),
659 select_and_scatter, /*desc=*/IrName(select_and_scatter, "init"),
666 llvm_ir::ForLoopNest source_loops(IrName(select_and_scatter), &ir_builder_);
687 llvm_ir::ForLoopNest window_loops(IrName(select_and_scatter, "window"),
[all...] |
| dot_op_emitter.cc | 764 llvm_ir::ForLoopNest loop_nest(llvm_ir::IrName(&dot_), ir_builder_);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| parallel_loop_emitter.cc | 111 llvm_ir::IrName(loop_name, "in_bounds"), ir_builder_, false);
|
| hlo_to_ir_bindings.cc | 154 ir_value->setName(llvm_ir::AsStringRef(llvm_ir::IrName(&hlo, "raw")));
155 typed_ir_value->setName(llvm_ir::AsStringRef(llvm_ir::IrName(&hlo, "typed")));
|
| elemental_ir_emitter.cc | 52 using llvm_ir::IrName;
367 llvm_ir::ForLoopNest loops(IrName(hlo), ir_builder_);
426 llvm_ir::ForLoopNest loops(IrName(hlo), ir_builder_);
|
| ir_emitter.cc | 50 using llvm_ir::IrName;
504 llvm_ir::ForLoopNest loop_nest(IrName(dot), &ir_builder_);
649 llvm_ir::ForLoopNest loops(IrName(reduce, "inner"), &ir_builder_);
738 .EmitLoop(IrName(random));
[all...] |
| ir_emitter_unnested.cc | 77 using llvm_ir::IrName;
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| elemental_ir_emitter.cc | 49 using llvm_ir::IrName;
[all...] |
