Home
last modified time | relevance | path

Searched refs:ir_arrays (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc820 std::vector<llvm_ir::IrArray> ir_arrays; in EmitPadToStaticFromMlir() local
824 mlir_input.extra_slice, &ir_arrays)); in EmitPadToStaticFromMlir()
826 const llvm_ir::IrArray source_array = ir_arrays[0]; in EmitPadToStaticFromMlir()
827 const llvm_ir::IrArray output_array = ir_arrays[1]; in EmitPadToStaticFromMlir()
829 absl::Span<const llvm_ir::IrArray>(ir_arrays).subspan(2); in EmitPadToStaticFromMlir()
947 std::vector<llvm_ir::IrArray> ir_arrays; in EmitSliceToDynamicFromMlir() local
951 mlir_input.extra_slice, &ir_arrays)); in EmitSliceToDynamicFromMlir()
973 const llvm_ir::IrArray data_array = ir_arrays.back(); in EmitSliceToDynamicFromMlir()
982 llvm::Value* source_buffer = ir_arrays[i].GetBasePointer(); in EmitSliceToDynamicFromMlir()
1041 ir_arrays[0].EmitReadArrayElement(dyn_index, &b_, /*name=*/"", in EmitSliceToDynamicFromMlir()
[all …]
Dir_emitter_unnested.h458 absl::Span<const llvm_ir::IrArray> ir_arrays,
658 std::vector<llvm_ir::IrArray>* ir_arrays);
663 std::vector<llvm_ir::IrArray>* ir_arrays);
668 std::vector<llvm_ir::IrArray>* ir_arrays);