| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_util.h | 90 string IrName(absl::string_view 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 | 169 return llvm_ir::IrName(prefix_, llvm_ir::IrName(name, suffix_)); in GetQualifiedName()
251 llvm_ir::IrName(suffix, absl::StrCat(dimension))); in AddLoopsForShapeOnDimensions()
|
| D | llvm_util.cc | 425 string IrName(absl::string_view a) { in IrName() function
431 string IrName(absl::string_view a, absl::string_view b) { in IrName() function
433 return IrName(absl::StrCat(a, ".", b)); in IrName()
435 return IrName(absl::StrCat(a, b)); in IrName()
438 string IrName(const HloInstruction* a, absl::string_view b) { in IrName() function
439 return IrName(a->name(), b); in IrName()
|
| D | dynamic_update_slice_util.cc | 233 fusion_output_array, launch_dimensions, IrName(dynamic_update_slice), b); in EmitFusedDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 85 using llvm_ir::IrName;
230 IrShapeType(bitcast->shape())->getPointerTo(), IrName(bitcast)); in HandleBitcast()
703 select_and_scatter, /*desc=*/IrName(select_and_scatter, "init"), in HandleSelectAndScatter()
710 llvm_ir::ForLoopNest source_loops(IrName(select_and_scatter), &b_); in HandleSelectAndScatter()
729 llvm_ir::ForLoopNest window_loops(IrName(select_and_scatter, "window"), &b_); in HandleSelectAndScatter()
1558 llvm_ir::ForLoopNest reduction_loop_nest(IrName(arg, "vectorized_inner"), in EmitInnerLoopForVectorizedReduction()
1702 llvm_ir::ForLoopNest loop_nest(IrName(reduce), &b_); in EmitVectorizedReduce()
1921 llvm_ir::ForLoopNest loops(IrName(slice), &b_); in HandleSlice()
1989 IrName(dynamic_update_slice, "in_place"), &b_); in HandleDynamicUpdateSlice()
2028 llvm_ir::ForLoopNest loops(IrName(pad, "assign"), &b_); in HandlePad()
[all …]
|
| D | dot_op_emitter.cc | 574 llvm_ir::ForLoopNest loop_nest(llvm_ir::IrName(dot_hlo_name_), b_); in EmitNaiveLlvmIrGemm()
1154 llvm_ir::IrName(&dot, "bdot"), /*start=*/0, /*end=*/batch_count, in EmitBatchDotOperation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_to_ir_bindings.cc | 146 ir_value->setName(llvm_ir::IrName(&hlo, "raw")); in GetTypedIrValue()
149 typed_ir_value->setName(llvm_ir::IrName(&hlo, "typed")); in GetTypedIrValue()
|
| D | parallel_loop_emitter.cc | 146 llvm_ir::IrName(loop_name, "in_bounds"), b_, false); in EmitIndexAndSetExitBasicBlock()
|
| D | elemental_ir_emitter.cc | 57 using llvm_ir::IrName;
|
| D | ir_emitter.cc | 74 using llvm_ir::IrName;
|
| D | ir_emitter_unnested.cc | 139 using llvm_ir::IrName;
912 llvm_ir::IrName(ir_name, "in_dyn_bounds"), &b_, false); in EmitPadToStaticFromMlir()
1033 llvm_ir::IrName(ir_name, "in_dyn_bounds"), &b_, false); in EmitSliceToDynamicFromMlir()
1927 IrName(mlir::GetNameFromLoc(fusion_op.getLoc())), in HandleFusion()
1961 desc.name = IrName(root); in HandleFusion()
3008 dimension_to_sort, values_arrays, IrName(context.name), xor_masks, &b_, in EmitSortFromMlir()
4868 IrName(name, StrCat("tile", id))); in EmitHlo021Tile()
5820 .EmitLoop(IrName(mlir::GetNameFromLoc(fusion.getLoc())), in EmitInputFusibleNonStridedSlices()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 55 using llvm_ir::IrName;
1769 /*insert_before=*/nullptr, IrName(hlo, "merge"), b_); in EmitElementalConcatenate()
1772 init_block->splitBasicBlock(b_->GetInsertPoint(), IrName(hlo, "merge")); in EmitElementalConcatenate()
1891 start_index_value->setName(IrName(hlo, StrCat("start_idx", i))); in EmitElementalDynamicSlice()
2064 start_index_value->setName(IrName(hlo, StrCat("start_idx", i))); in EmitElementalDynamicUpdateSlice()
2192 IrName(hlo, "inner"), index_typed_const(0), in EmitElementalDot()
2584 llvm_ir::IrName(map_instr))); in EmitElementalMap()
2633 llvm_ir::ForLoopNest loops(IrName(reduce_window), b_, index_type); in EmitElementalReduceWindow()
2749 llvm_ir::ForLoopNest loops(IrName(reduce, "inner"), b(), index_type); in EmitElementalReduce()
2851 llvm_ir::ForLoopNest loops(IrName(convolution, "inner"), b_); in EmitConvolution()
|