| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 77 using llvm_ir::IrName;
171 IrShapeType(bitcast->shape())->getPointerTo(), IrName(bitcast)); in HandleBitcast()
657 llvm_ir::ForLoopNest loops(IrName(reduce_window, "inner"), &b_); in EmitElementalReduceWindow()
784 select_and_scatter, /*desc=*/IrName(select_and_scatter, "init"), in HandleSelectAndScatter()
791 llvm_ir::ForLoopNest source_loops(IrName(select_and_scatter), &b_); in HandleSelectAndScatter()
810 llvm_ir::ForLoopNest window_loops(IrName(select_and_scatter, "window"), &b_); in HandleSelectAndScatter()
980 llvm_ir::ForLoopNest loops(IrName(convolution, "inner"), &b_); in EmitElementalConvolution()
1593 llvm_ir::ForLoopNest reduction_loop_nest(IrName(arg, "vectorized_inner"), in EmitInnerLoopForVectorizedReduction()
1722 llvm_ir::ForLoopNest loop_nest(IrName(reduce), &b_); in EmitVectorizedReduce()
1837 llvm_ir::ForLoopNest loops(IrName(reduce, "inner"), &b_); in EmitElementalReduce()
[all …]
|
| D | elemental_ir_emitter.cc | 121 operands, llvm_ir::IrName(hlo)); in MakeElementGenerator()
|
| D | dot_op_emitter.cc | 431 llvm_ir::ForLoopNest loop_nest(llvm_ir::IrName(dot_hlo_name_), b_); in EmitNaiveLlvmIrGemm()
958 llvm_ir::IrName(&dot, "bdot"), /*start=*/0, /*end=*/batch_count, in EmitBatchDotOperation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_util.h | 90 string IrName(string a);
91 string IrName(absl::string_view a, absl::string_view b);
92 string IrName(const HloInstruction* a, absl::string_view b = "");
|
| D | llvm_loop.cc | 168 return llvm_ir::IrName(prefix_, llvm_ir::IrName(name, suffix_)); in GetQualifiedName()
250 llvm_ir::IrName(suffix, absl::StrCat(dimension))); in AddLoopsForShapeOnDimensions()
|
| D | llvm_util.cc | 423 string IrName(string a) { in IrName() function
428 string IrName(absl::string_view a, absl::string_view b) { in IrName() function
430 return IrName(absl::StrCat(a, ".", b)); in IrName()
432 return IrName(absl::StrCat(a, b)); in IrName()
435 string IrName(const HloInstruction* a, absl::string_view b) { in IrName() function
436 return IrName(a->name(), b); in IrName()
|
| D | dynamic_update_slice_util.cc | 182 fusion_output_array, launch_dimensions, IrName(fusion), b); in EmitFusedDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_to_ir_bindings.cc | 168 ir_value->setName(llvm_ir::IrName(&hlo, "raw")); in GetTypedIrValue()
171 typed_ir_value->setName(llvm_ir::IrName(&hlo, "typed")); in GetTypedIrValue()
|
| D | elemental_ir_emitter.cc | 55 using llvm_ir::IrName;
377 llvm_ir::ForLoopNest loops(IrName(hlo), b_, index_type); in MakeElementGenerator()
455 llvm_ir::ForLoopNest loops(IrName(hlo), b_, index_type); in MakeElementGenerator()
|
| D | parallel_loop_emitter.cc | 136 llvm_ir::IrName(loop_name, "in_bounds"), b_, false); in EmitIndexAndSetExitBasicBlock()
|
| D | ir_emitter.cc | 52 using llvm_ir::IrName;
530 llvm_ir::ForLoopNest loop_nest(IrName(dot), &b_); in HandleDot()
688 llvm_ir::ForLoopNest loops(IrName(reduce, "inner"), &b_); in HandleReduce()
|
| D | ir_emitter_unnested.cc | 106 using llvm_ir::IrName;
851 llvm_ir::ForLoopNest window_loops(IrName(select_and_scatter, "inner"), &b_, in HandleSelectAndScatter()
974 .EmitLoop(IrName(select_and_scatter), index_type); in HandleSelectAndScatter()
1200 .EmitLoop(IrName(scatter), in EmitScatter()
1334 dimension_to_sort, values_arrays, IrName(sort), xor_masks, &b_, in HandleSort()
2010 .EmitLoop(IrName(hlo))); in BuildInitializerThunk()
2020 .EmitLoop(IrName(hlo))); in BuildInitializerThunk()
2194 IrName(&hlo), in EmitTargetElementLoopInThunk()
2211 .EmitLoop(IrName(&hlo), in EmitTargetElementLoopInThunk()
3066 IrName(unnested_hlo, StrCat("tile", id))); in EmitKernel()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 53 using llvm_ir::IrName;
1722 /*insert_before=*/nullptr, IrName(hlo, "merge"), b_); in EmitElementalConcatenate()
1725 init_block->splitBasicBlock(b_->GetInsertPoint(), IrName(hlo, "merge")); in EmitElementalConcatenate()
1844 start_index_value->setName(IrName(hlo, StrCat("start_idx", i))); in EmitElementalDynamicSlice()
2005 start_index_value->setName(IrName(hlo, StrCat("start_idx", i))); in EmitElementalDynamicUpdateSlice()
2133 IrName(hlo, "inner"), index_typed_const(0), in EmitElementalDot()
|