Home
last modified time | relevance | path

Searched full:mlir (Results 1 – 25 of 2612) sorted by relevance

12345678910>>...105

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/
DBUILD5 load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "gentbl_filegroup", "td_library")
13 "include/mlir-hlo/Dialect/mhlo/IR/chlo_ops.td",
14 "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td",
15 "include/mlir-hlo/Dialect/lhlo/IR/lhlo_ops.td",
23 srcs = glob(["include/mlir-hlo/Dialect/mhlo/IR/*.td"]),
28 srcs = glob(["include/mlir-hlo/Dialect/mhlo/IR/*.td"]),
32 "@llvm-project//mlir:BuiltinDialectTdFiles",
33 "@llvm-project//mlir:ControlFlowInterfacesTdFiles",
34 "@llvm-project//mlir:CopyOpInterfaceTdFiles",
35 "@llvm-project//mlir:InferTypeOpInterfaceTdFiles",
[all …]
/external/tensorflow/tensorflow/compiler/mlir/python/mlir_wrapper/
Dops.cc16 #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project
17 #include "mlir/IR/BuiltinOps.h" // from @llvm-project
18 #include "mlir/IR/Operation.h" // from @llvm-project
19 #include "tensorflow/compiler/mlir/python/mlir_wrapper/mlir_wrapper.h"
20 #include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h"
23 py::class_<mlir::Operation, std::unique_ptr<mlir::Operation, py::nodelete>>( in init_ops()
25 .def("getRegion", &mlir::Operation::getRegion, in init_ops()
27 .def("getResult", &mlir::Operation::getResult) in init_ops()
28 .def("dump", &mlir::Operation::dump) in init_ops()
29 .def("getNumResults", &mlir::Operation::getNumResults); in init_ops()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
DBUILD1 load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library")
13 default_visibility = ["//tensorflow/compiler/mlir/tools/kernel_gen:friends"],
24 "@llvm-project//mlir:FuncDialect",
25 "@llvm-project//mlir:IR",
26 "@llvm-project//mlir:LLVMDialect",
27 "@llvm-project//mlir:Transforms",
38 "//tensorflow/compiler/mlir/tools/kernel_gen/ir:tf_framework_ops",
40 "@llvm-project//mlir:FuncDialect",
41 "@llvm-project//mlir:FuncToLLVM",
42 "@llvm-project//mlir:IR",
[all …]
/external/tensorflow/tensorflow/dtensor/mlir/
Dvalue_utils.cc16 #include "tensorflow/dtensor/mlir/value_utils.h"
18 #include "mlir/IR/BuiltinAttributes.h" // from @llvm-project
19 #include "mlir/IR/BuiltinTypes.h" // from @llvm-project
20 #include "tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.h"
22 #include "tensorflow/dtensor/mlir/ir/tf_dtensor.h"
23 #include "tensorflow/dtensor/mlir/op_utils.h"
29 // Given a mlir::Value will trace the value back through
32 mlir::Value GetForwardedInput(mlir::Value value) { in GetForwardedInput()
36 if (mlir::BlockArgument argument = value.dyn_cast<mlir::BlockArgument>()) { in GetForwardedInput()
37 mlir::Region* region = argument.getParentRegion(); in GetForwardedInput()
[all …]
Dcreate_dtensor_mlir_passes.h21 #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project
22 #include "mlir/IR/BuiltinOps.h" // from @llvm-project
23 #include "mlir/Pass/Pass.h" // from @llvm-project
24 #include "mlir/Pass/PassManager.h" // from @llvm-project
25 #include "tensorflow/compiler/mlir/tensorflow/transforms/passes.h"
31 std::unique_ptr<mlir::OperationPass<mlir::func::FuncOp>>
34 std::unique_ptr<mlir::OperationPass<mlir::func::FuncOp>>
37 std::unique_ptr<mlir::OperationPass<mlir::func::FuncOp>> CreateDTensorDCE();
39 std::unique_ptr<mlir::OperationPass<mlir::func::FuncOp>>
42 std::unique_ptr<mlir::OperationPass<mlir::func::FuncOp>>
[all …]
Dmesh_propagation.cc23 #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project
24 #include "mlir/IR/Attributes.h" // from @llvm-project
25 #include "mlir/IR/Builders.h" // from @llvm-project
26 #include "mlir/IR/BuiltinOps.h" // from @llvm-project
27 #include "mlir/IR/Diagnostics.h" // from @llvm-project
28 #include "mlir/IR/Operation.h" // from @llvm-project
29 #include "mlir/IR/Value.h" // from @llvm-project
30 #include "mlir/IR/Visitors.h" // from @llvm-project
31 #include "mlir/Pass/Pass.h" // from @llvm-project
32 #include "mlir/Pass/PassManager.h" // from @llvm-project
[all …]
Dmerge_clusters.cc26 #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project
27 #include "mlir/IR/Attributes.h" // from @llvm-project
28 #include "mlir/IR/Builders.h" // from @llvm-project
29 #include "mlir/IR/BuiltinOps.h" // from @llvm-project
30 #include "mlir/IR/Diagnostics.h" // from @llvm-project
31 #include "mlir/IR/Operation.h" // from @llvm-project
32 #include "mlir/IR/Types.h" // from @llvm-project
33 #include "mlir/IR/Value.h" // from @llvm-project
34 #include "mlir/IR/Visitors.h" // from @llvm-project
35 #include "mlir/Pass/Pass.h" // from @llvm-project
[all …]
Dhandle_cross_cluster_dependencies.cc21 #include "mlir/IR/Block.h" // from @llvm-project
22 #include "mlir/IR/Builders.h" // from @llvm-project
23 #include "mlir/IR/BuiltinOps.h" // from @llvm-project
24 #include "mlir/IR/BuiltinTypes.h" // from @llvm-project
25 #include "mlir/IR/Operation.h" // from @llvm-project
26 #include "mlir/IR/UseDefLists.h" // from @llvm-project
27 #include "mlir/IR/Value.h" // from @llvm-project
28 #include "mlir/Support/LogicalResult.h" // from @llvm-project
29 #include "mlir/Transforms/Passes.h" // from @llvm-project
30 #include "mlir/Transforms/RegionUtils.h" // from @llvm-project
[all …]
DBUILD1 # MLIR passes for DTensor support.
5 load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library")
11 # Allow visibility from the mlir language server.
12 "//learning/brain/mlir/mlir_lsp_server:__pkg__",
30 tblgen = "@llvm-project//mlir:mlir-tblgen",
33 "//tensorflow/compiler/mlir/tensorflow:ir/tf_op_base.td",
34 "//tensorflow/compiler/mlir/tensorflow:ir/tf_op_interfaces.td",
37 "@llvm-project//mlir:CallInterfacesTdFiles",
38 "@llvm-project//mlir:FuncTdFiles",
39 "@llvm-project//mlir:InferTypeOpInterfaceTdFiles",
[all …]
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/
Dquantize_model.cc15 #include "tensorflow/compiler/mlir/quantization/tensorflow/python/quantize_model.h"
26 #include "mlir/Dialect/Arithmetic/IR/Arithmetic.h" // from @llvm-project
27 #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project
28 #include "mlir/Dialect/Quant/QuantOps.h" // from @llvm-project
29 #include "mlir/Dialect/SCF/IR/SCF.h" // from @llvm-project
30 #include "mlir/Dialect/Shape/IR/Shape.h" // from @llvm-project
31 #include "mlir/IR/BuiltinOps.h" // from @llvm-project
32 #include "mlir/IR/MLIRContext.h" // from @llvm-project
33 #include "mlir/IR/OwningOpRef.h" // from @llvm-project
34 #include "mlir/Pass/Pass.h" // from @llvm-project
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/
DBUILD8 load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")
23 "//learning/brain/experimental/mlir/tflite/tfmrt/...",
24 "//learning/brain/mlir/...",
27 "//tensorflow/compiler/mlir/...",
29 "//waymo/ml/compiler/mlir/...",
30 # Allow visibility from the mlir language server.
31 "//learning/brain/mlir/mlir_lsp_server/...",
33 "//learning/brain/research/babelfish/inference/speech_tflite/mlir/...",
46 "//tensorflow/compiler/mlir/lite/quantization:quantization_td_files",
47 "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops_td_files",
[all …]
Dtf_tfl_passes.cc16 #include "tensorflow/compiler/mlir/lite/tf_tfl_passes.h"
22 #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project
23 #include "mlir/IR/Attributes.h" // from @llvm-project
24 #include "mlir/IR/BuiltinOps.h" // from @llvm-project
25 #include "mlir/Pass/Pass.h" // from @llvm-project
26 #include "mlir/Pass/PassManager.h" // from @llvm-project
27 #include "mlir/Transforms/Passes.h" // from @llvm-project
28 #include "tensorflow/compiler/mlir/lite/quantization/quantization_config.h"
29 #include "tensorflow/compiler/mlir/lite/quantization/quantization_passes.h"
30 #include "tensorflow/compiler/mlir/lite/quantization/tensorflow/passes.h"
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/
DBUILD6 "@llvm-project//mlir:tblgen.bzl",
34 "//tensorflow/compiler/mlir/lite:tensorflow_lite",
35 "//tensorflow/compiler/mlir/lite/experimental/tac/hardwares:target_hardware",
36 "//tensorflow/compiler/mlir/tensorflow",
39 "@llvm-project//mlir:ArithmeticDialect",
40 "@llvm-project//mlir:FuncDialect",
41 "@llvm-project//mlir:IR",
53 "//tensorflow/compiler/mlir/lite:flatbuffer_translate_lib",
54 "//tensorflow/compiler/mlir/lite:tensorflow_lite",
55 "//tensorflow/compiler/mlir/lite/experimental/tac/hardwares:all-target-hardwares",
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
Dtf_cpurt_passes.cc16 #include "tensorflow/compiler/mlir/tfrt/jit/tf_cpurt_passes.h"
22 #include "mlir/Conversion/ShapeToStandard/ShapeToStandard.h"
23 #include "mlir/Dialect/Async/IR/Async.h"
24 #include "mlir/Dialect/Linalg/Passes.h"
25 #include "mlir/Dialect/Linalg/Transforms/CodegenStrategy.h"
26 #include "mlir/Dialect/MemRef/IR/MemRef.h"
27 #include "mlir/Dialect/MemRef/Transforms/Passes.h"
28 #include "mlir/Dialect/StandardOps/IR/Ops.h"
29 #include "mlir/Dialect/StandardOps/Transforms/Passes.h"
30 #include "mlir/Dialect/Tensor/Transforms/Passes.h"
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.cc18 #include "tensorflow/compiler/mlir/xla/hlo_utils.h"
20 #include "mlir/IR/AffineMap.h" // from @llvm-project
21 #include "mlir/IR/Attributes.h" // from @llvm-project
22 #include "mlir/IR/BuiltinTypes.h" // from @llvm-project
23 #include "mlir/IR/TypeUtilities.h" // from @llvm-project
25 #include "tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo/IR/lhlo_ops.h"
32 using mlir::AffineMap;
33 using mlir::Builder;
34 using mlir::DenseElementsAttr;
35 using mlir::ShapedType;
[all …]
Dhlo_function_importer.h22 #include "mlir/Dialect/Arithmetic/IR/Arithmetic.h" // from @llvm-project
23 #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project
24 #include "mlir/IR/Attributes.h" // from @llvm-project
25 #include "mlir/IR/Builders.h" // from @llvm-project
26 #include "mlir/IR/BuiltinOps.h" // from @llvm-project
27 #include "mlir/IR/BuiltinTypes.h" // from @llvm-project
28 #include "mlir/IR/MLIRContext.h" // from @llvm-project
29 #include "tensorflow/compiler/mlir/tensorflow/utils/error_util.h"
31 #include "tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.h"
44 // HLO bounded dynamic shapes can be converted to either MLIR dynamic shapes
[all …]
DBUILD2 load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")
20 tblgen = "@llvm-project//mlir:mlir-tblgen",
23 "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops_td_files",
26 "@llvm-project//mlir:FuncTdFiles",
27 "@llvm-project//mlir:TensorOpsTdFiles",
43 tblgen = "@llvm-project//mlir:mlir-tblgen",
46 "@llvm-project//mlir:PassBaseTdFiles",
62 tblgen = "@llvm-project//mlir:mlir-tblgen",
65 "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops_td_files",
68 "@llvm-project//mlir:FuncTdFiles",
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tfrt/
DBUILD1 load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")
31 "//learning/brain/experimental/mlir/tflite/tfmrt/...",
32 "//learning/brain/experimental/mlir/tfrt_compiler/...",
36 "//learning/serving/contrib/tfrt/mlir/...",
37 # Allow visibility from the mlir language server.
38 "//learning/brain/mlir/mlir_lsp_server/...",
52 "//tensorflow/compiler/mlir/tfrt/ir:tfrt_fallback_td_files",
53 "@llvm-project//mlir:OpBaseTdFiles",
54 "@llvm-project//mlir:SideEffectInterfacesTdFiles",
73 tblgen = "@llvm-project//mlir:mlir-tblgen",
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dkernel_creator.cc22 #include "tensorflow/compiler/mlir/tools/kernel_gen/kernel_creator.h"
26 #include "mlir/Conversion/AffineToStandard/AffineToStandard.h" // from @llvm-project
27 #include "mlir/Conversion/ComplexToStandard/ComplexToStandard.h" // from @llvm-project
28 #include "mlir/Conversion/FuncToLLVM/ConvertFuncToLLVMPass.h" // from @llvm-project
29 #include "mlir/Conversion/GPUCommon/GPUCommonPass.h" // from @llvm-project
30 #include "mlir/Conversion/GPUToNVVM/GPUToNVVMPass.h" // from @llvm-project
31 #include "mlir/Conversion/ReconcileUnrealizedCasts/ReconcileUnrealizedCasts.h" // from @llvm-proje…
32 #include "mlir/Conversion/SCFToControlFlow/SCFToControlFlow.h" // from @llvm-project
33 #include "mlir/Conversion/SCFToGPU/SCFToGPUPass.h" // from @llvm-project
34 #include "mlir/Conversion/ShapeToStandard/ShapeToStandard.h" // from @llvm-project
[all …]
/external/tensorflow/tensorflow/compiler/mlir/
DBUILD2 # TensorFlow/TensorFlow Lite/XLA MLIR dialects and tools.
19 packages = ["//tensorflow/compiler/mlir/..."],
54 "@llvm-project//mlir:IR",
66 "@llvm-project//mlir:IR",
76 "//tensorflow/compiler/mlir/lite:tensorflow_lite",
77 "//tensorflow/compiler/mlir/lite:tf_tfl_passes",
78 "//tensorflow/compiler/mlir/tensorflow",
79 "//tensorflow/compiler/mlir/tensorflow:compile_mlir_util",
80 "//tensorflow/compiler/mlir/tensorflow:tensorflow_passes",
81 "//tensorflow/compiler/mlir/tensorflow:tensorflow_test_passes",
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tfr/
DBUILD13 load("//tensorflow/compiler/mlir/tfr:build_defs.bzl", "gen_op_libraries")
15 "@llvm-project//mlir:tblgen.bzl",
19 load("//tensorflow/compiler/mlir:glob_lit_test.bzl", "glob_lit_tests")
34 # Allow visibility from the mlir language server.
35 "//learning/brain/mlir/mlir_lsp_server/...",
46 "//tensorflow/compiler/mlir/lite/quantization/ir:QuantizationOpsTdFiles",
47 "//tensorflow/compiler/mlir/tensorflow:tensorflow_ops_td_files",
48 "@llvm-project//mlir:CallInterfacesTdFiles",
49 "@llvm-project//mlir:ControlFlowInterfacesTdFiles",
50 "@llvm-project//mlir:FunctionInterfacesTdFiles",
[all …]
/external/tensorflow/tensorflow/dtensor/mlir/utils/
Dcollective_lowering.cc25 #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project
26 #include "mlir/IR/Attributes.h" // from @llvm-project
27 #include "mlir/IR/Builders.h" // from @llvm-project
28 #include "mlir/IR/BuiltinAttributes.h" // from @llvm-project
29 #include "mlir/IR/BuiltinOps.h" // from @llvm-project
30 #include "mlir/IR/BuiltinTypes.h" // from @llvm-project
31 #include "mlir/IR/MLIRContext.h" // from @llvm-project
32 #include "mlir/IR/Operation.h" // from @llvm-project
33 #include "mlir/IR/Types.h" // from @llvm-project
34 #include "mlir/IR/Value.h" // from @llvm-project
[all …]
/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/
Dcustom_call_encoding.h28 #include "mlir/Dialect/LLVMIR/LLVMDialect.h" // from @llvm-project
29 #include "mlir/IR/Attributes.h" // from @llvm-project
30 #include "mlir/IR/BuiltinOps.h" // from @llvm-project
31 #include "mlir/IR/ImplicitLocOpBuilder.h" // from @llvm-project
32 #include "mlir/IR/SymbolTable.h" // from @llvm-project
33 #include "mlir/Support/LogicalResult.h" // from @llvm-project
75 mlir::Value type_id; // !llvm.ptr<i64>
76 mlir::Value value; // !llvm.ptr<ArgType>
81 virtual mlir::LogicalResult Match(mlir::Value value,
82 mlir::Value conterted) const = 0;
[all …]
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/
DBUILD4 load("//tensorflow/compiler/mlir/quantization/tensorflow:internal_visibility_allowlist.bzl", "inter…
5 load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "td_library")
10 "//tensorflow/compiler/mlir/quantization/...",
11 "//tensorflow/compiler/mlir/lite/...",
35 "passes/quantized_function_library.mlir",
36 "passes/quantized_function_library_uniform_quantized_drq.mlir",
37 "passes/quantized_function_library_tf_drq.mlir",
57 "//tensorflow/compiler/mlir/lite/quantization:quantization_lib",
58 "//tensorflow/compiler/mlir/quantization/tensorflow:quantization_options_proto_cc",
59 "//tensorflow/compiler/mlir/tensorflow",
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
Dtf_jitrt_passes.h22 #include "mlir/Dialect/Func/IR/FuncOps.h"
23 #include "mlir/Dialect/LLVMIR/LLVMDialect.h"
24 #include "mlir/Dialect/Linalg/IR/Linalg.h"
25 #include "mlir/Dialect/MemRef/IR/MemRef.h"
26 #include "mlir/Dialect/SCF/IR/SCF.h"
27 #include "mlir/Dialect/Vector/IR/VectorOps.h"
28 #include "mlir/Pass/Pass.h"
29 #include "mlir/Pass/PassManager.h"
30 #include "tensorflow/compiler/mlir/tensorflow/ir/tf_device.h"
31 #include "tensorflow/compiler/mlir/tensorflow/ir/tf_dialect.h"
[all …]

12345678910>>...105