| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | kernel_creator.cc | 70 namespace kernel_gen { namespace
225 pm.addPass(mlir::kernel_gen::transforms::CreateHloBufferizePass()); in LowerTFtoLoops()
231 mlir::kernel_gen::transforms::CreateBufferReusePass()); in LowerTFtoLoops()
267 mlir::kernel_gen::transforms::CreateMapParallelLoopsPass()); in LowerLoopsToGPUorCPU()
274 pm.addPass(mlir::kernel_gen::transforms::CreateShapeToDescriptorsPass()); in LowerLoopsToGPUorCPU()
289 pm.addPass(mlir::kernel_gen::tf_framework:: in LowerLoopsToGPUorCPU()
293 pm.addPass(mlir::kernel_gen::transforms::CreateFinalBufferizePass()); in LowerLoopsToGPUorCPU()
332 mlir::kernel_gen::tf_framework::CreateEmbedTFFrameworkAssertPass()); in LowerLoopsToGPUorCPU()
335 mlir::kernel_gen::transforms::CreateEmbedMemRefPrintsPass()); in LowerLoopsToGPUorCPU()
364 kernelPm.addPass(mlir::kernel_gen::transforms::CreateGpuKernelToRocdlPass()); in LowerKernelBodiesToLowLevelIr()
[all …]
|
| D | BUILD | 51 "//tensorflow/compiler/mlir/tools/kernel_gen/transforms:passes",
98 "//tensorflow/compiler/mlir/tools/kernel_gen/tests/tf_to_kernel:__pkg__",
132 visibility = ["//tensorflow/compiler/mlir/tools/kernel_gen/tests:__subpackages__"],
137 "//tensorflow/compiler/mlir/tools/kernel_gen/ir:tf_framework_ops",
138 "//tensorflow/compiler/mlir/tools/kernel_gen/transforms:passes",
155 "//tensorflow/compiler/mlir/tools/kernel_gen/ir:tf_framework_ops",
|
| D | crash_handler.h | 22 namespace kernel_gen {
|
| D | tf_framework_c_interface.h | 22 namespace kernel_gen {
|
| D | kernel_creator.h | 34 namespace kernel_gen {
|
| D | tf_to_kernel.cc | 47 namespace kernel_gen { namespace
178 auto status = tensorflow::kernel_gen::Run( in main()
|
| D | tf_framework_c_interface.cc | 23 namespace kernel_gen { namespace
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | embed_tf_framework_pass.cc | 24 namespace kernel_gen { namespace
44 registry.insert<mlir::kernel_gen::tf_framework::TFFrameworkDialect>(); in getDependentDialects()
85 registry.insert<mlir::kernel_gen::tf_framework::TFFrameworkDialect>(); in getDependentDialects()
|
| D | BUILD | 13 default_visibility = ["//tensorflow/compiler/mlir/tools/kernel_gen:friends"],
23 "//tensorflow/compiler/mlir/tools/kernel_gen/ir:tf_framework_ops",
55 "//tensorflow/compiler/mlir/tools/kernel_gen/ir:tf_framework_ops",
136 "//tensorflow/compiler/mlir/tools/kernel_gen/ir:tf_framework_ops",
|
| D | buffer_reuse_pass.cc | 44 mlir::kernel_gen::tf_framework::TFAllocOp::kReuseOutputAttrName;
46 mlir::kernel_gen::tf_framework::TFAllocOp::kReuseInputCandidatesAttrName;
48 mlir::kernel_gen::tf_framework::TFFrameworkDialect::kTFEntryAttrName;
51 namespace kernel_gen { namespace
|
| D | map_parallel_loops_to_gpu.cc | 20 namespace kernel_gen { namespace
|
| D | fuse_inner_parallel_loops_pass.cc | 21 namespace kernel_gen { namespace
|
| D | rewriters.h | 29 namespace kernel_gen {
|
| D | parallel_loops_to_sequential.cc | 22 namespace kernel_gen { namespace
|
| D | shape_to_descriptors_pass.cc | 32 namespace kernel_gen { namespace
|
| D | passes.h | 27 namespace kernel_gen {
|
| D | kernel_lowering_passes.cc | 29 namespace kernel_gen { namespace
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tools/kernel-gen-opt/ |
| D | kernel-gen-opt.cc | 29 mlir::kernel_gen::registerKernelGenPasses(); in main()
35 registry.insert<mlir::kernel_gen::tf_framework::TFFrameworkDialect>(); in main()
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | build_defs.bzl | 186 default = Label("//tensorflow/compiler/mlir/tools/kernel_gen:tf_to_kernel"),
271 "$(location //tensorflow/compiler/mlir/tools/kernel_gen:tf_to_kernel)",
288 "//tensorflow/compiler/mlir/tools/kernel_gen:tf_to_kernel",
300 ] + ["//tensorflow/compiler/mlir/tools/kernel_gen:tf_framework_c_interface"]
305 "//tensorflow/compiler/mlir/tools/kernel_gen:tf_gpu_runtime_wrappers",
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ir/ |
| D | tf_framework_ops.td | 23 include "tensorflow/compiler/mlir/tools/kernel_gen/ir/tf_status.td"
33 let cppNamespace = "::mlir::kernel_gen::tf_framework";
42 CPred<"$_self.isa<::mlir::kernel_gen::tf_framework::OpKernelContextType>()">,
44 BuildableType<"$_builder.getType<::mlir::kernel_gen::tf_framework::OpKernelContextType>()"> {
|
| D | tf_framework_ops.h | 32 namespace kernel_gen {
|
| D | BUILD | 6 default_visibility = ["//tensorflow/compiler/mlir/tools/kernel_gen:friends"],
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/tf_to_kernel/ |
| D | BUILD | 7 "//tensorflow/compiler/mlir/tools/kernel_gen:tf_to_kernel",
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | tf_mlir_opt_main.cc | 43 registry.insert<mlir::kernel_gen::tf_framework::TFFrameworkDialect>(); in main()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/ |
| D | BUILD | 22 "//tensorflow/compiler/mlir/tools/kernel_gen:kernel-gen-opt",
|