| /external/tensorflow/tensorflow/python/compiler/mlir/ |
| D | mlir_test.py | 35 mlir_module = mlir.convert_graph_def('')
37 self.assertIn('func @main', mlir_module)
56 mlir_module = mlir.convert_function(concrete_function, show_debug_info=True)
57 self.assertRegex(mlir_module, r'func @.*sqr.*\(')
58 self.assertRegex(mlir_module, r'callsite\(".*mlir_test.py":')
73 mlir_module = mlir.convert_function(concrete_function)
74 self.assertRegex(mlir_module, r'func @.*caller.*\(')
75 self.assertRegex(mlir_module, r'func private @.*callee.*\(')
85 mlir_module = mlir.convert_function(concrete_function, pass_pipeline='')
86 self.assertRegex(mlir_module, r'tf\.PrintV2')
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter_test.cc | 52 mlir::OwningModuleRef mlir_module( in CompileHloConvAndGetMlir() local
58 mlir_module->push_back(function); in CompileHloConvAndGetMlir()
59 (void)mlir_module->verify(); in CompileHloConvAndGetMlir()
69 mlir::PassManager pm(mlir_module->getContext()); in CompileHloConvAndGetMlir()
73 CHECK(mlir::succeeded(pm.run(*mlir_module))); in CompileHloConvAndGetMlir()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | serialize_mlir_module_utils.cc | 38 mlir::OwningModuleRef* mlir_module) { in DeserializeMlirModule() argument
41 TF_RET_CHECK(mlir_module) << "unexpected null MLIR module pointer"; in DeserializeMlirModule()
48 *mlir_module = mlir::parseSourceString(serialized_mlir_module, mlir_context); in DeserializeMlirModule()
49 if (!*mlir_module) in DeserializeMlirModule()
|
| D | serialize_mlir_module_utils.h | 35 mlir::OwningModuleRef* mlir_module);
|
| D | compile_mlir_util.cc | 500 mlir::OwningModuleRef mlir_module; in CompileSerializedMlirToXlaHlo() local
503 DeserializeMlirModule(mlir_module_string, &mlir_context, &mlir_module)); in CompileSerializedMlirToXlaHlo()
509 mlir_module.get(), tensor_or_resource_shapes, device_type, use_tuple_args, in CompileSerializedMlirToXlaHlo()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tpu_compile_op_replication_pass.mlir | 5 … = "/job:worker/replica:0/task:1/device:CPU:0", metadata = "metadata", mlir_module = "mlir_module"}
7 … = "/job:worker/replica:0/task:0/device:CPU:0", metadata = "metadata", mlir_module = "mlir_module"}
8 …:worker/replica:0/task:0/device:CPU:0", metadata = "metadata", mlir_module = "mlir_module"} : () -…
|
| D | tpu-dynamic-layout-pass.mlir | 14 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
52 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
88 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
119 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
151 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
184 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
215 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
249 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
294 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
332 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
[all …]
|
| D | materialize_passthrough_op.mlir | 13 …T", "tfdtype$DT_FLOAT"], Toutputs = ["tfdtype$DT_FLOAT"], device = "", mlir_module = "\0Afunc @mai…
|
| D | tpu-merge-variables-with-execute.mlir | 23 // CHECK-SAME: mlir_module
28 …mlir_module = "module attributes {tf.versions = {producer = 888 : i32}} {\0A func @main(%arg0: te…
68 // CHECK-SAME: mlir_module
72 …mlir_module = "module attributes {tf.versions = {producer = 888 : i32}} {\0A func @main(%arg0: te…
131 // CHECK-SAME: mlir_module
135 …mlir_module = "module attributes {tf.versions = {producer = 888 : i32}} {\0A func @main(%arg0: te…
|
| D | tpu-variable-runtime-reformatting.mlir | 46 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
129 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
193 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
266 mlir_module = "..."} : () -> (tensor<!tf.string>, tensor<2x!tf.string>)
|
| D | tpu_rewrite.mlir | 741 // CHECK-SAME: mlir_module
788 // CHECK-SAME: mlir_module
859 // CHECK-SAME: mlir_module
908 // CHECK-SAME: mlir_module
953 // CHECK-SAME: mlir_module
1006 // CHECK-SAME: mlir_module
1053 // CHECK-SAME: mlir_module
1067 // CHECK-SAME: mlir_module
1109 // CHECK-SAME: mlir_module
1123 // CHECK-SAME: mlir_module
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_impl.cc | 60 std::string mlir_module; in CreateMlirImpl() local
62 ctx, &metadata, /*function_name=*/nullptr, &mlir_module)); in CreateMlirImpl()
65 mlir_module, metadata, metadata.num_cores_per_replica(), in CreateMlirImpl()
|
| D | tpu_compile_op_impl.h | 38 TpuCompileOpKernelImpl(const std::string& mlir_module, in TpuCompileOpKernelImpl() argument
42 : TpuCompileOpKernelCommon(mlir_module, metadata, num_computations, in TpuCompileOpKernelImpl()
|
| D | tpu_compile_op_support.h | 50 const std::string& mlir_module; member
150 std::string* mlir_module);
|
| D | tpu_compile_op_common.h | 68 TpuCompileOpKernelCommon(const std::string& mlir_module, in TpuCompileOpKernelCommon() argument
74 mlir_module_(mlir_module), in TpuCompileOpKernelCommon()
|
| D | tpu_compile_op_support.cc | 348 *compilation_request.mutable_mlir_module() = mlir_computation.mlir_module; in CreateTpuCompilationRequest()
390 std::string* mlir_module) { in CompileOpMetadataFromContext() argument
406 if (mlir_module != nullptr) { in CompileOpMetadataFromContext()
407 TF_RETURN_IF_ERROR(ctx->GetAttr("mlir_module", mlir_module)); in CompileOpMetadataFromContext()
|
| D | tpu_compile.proto | 51 string mlir_module = 8; field
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | mlir_emitter.cc | 114 mlir::OwningModuleRef mlir_module = mlir::ModuleOp::create(loc); in EmitMlirFuncAndCall() local
115 mlir_module->push_back(function); in EmitMlirFuncAndCall()
121 MakeLLVMModule(std::move(mlir_module), &b->getContext()); in EmitMlirFuncAndCall()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_MlirPassthroughOp.pbtxt | 21 mlir_module = '''python
30 return mlir_passthrough_op([x, y], mlir_module, Toutputs=[tf.float32])
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | mlir_passthrough_op.pbtxt | 4 # CHECK: mlir_module = "\0Afunc @main(%arg0 : tensor<10xf32>, %arg1 : tensor<10xf32>) -> tensor<10x…
81 key: "mlir_module"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | MlirPassthroughOp.pbtxt | 12 name: "mlir_module"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | MlirPassthroughOp.pbtxt | 12 name: "mlir_module"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
| D | type_attr.mlir | 29 …sor<16xf32>], Tinputs = [tensor<16xf32>], Toutputs = [tensor<16xf32>], mlir_module = ""} : (tensor…
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | materialize_mlir_passthrough_op.cc | 47 std::string module_string(passthrough_op.mlir_module()); in runOnFunction()
|
| D | tpu_merge_variables_with_execute.cc | 448 compile.mlir_module().str(), compile.getContext(), &module_ref); in UpdateSerializedModule()
|