| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | shape_inference_utils.cc | 32 assert(op->getName().getDialectNamespace() == in InferReturnTypeComponentsForTFOp()
33 TensorFlowDialect::getDialectNamespace()); in InferReturnTypeComponentsForTFOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | order_by_dialect.cc | 102 bool a_current = a->getName().getDialectNamespace() == current_dialect; in groupOperationsByDialect()
103 bool b_current = b->getName().getDialectNamespace() == current_dialect; in groupOperationsByDialect()
119 current_dialect = best->getName().getDialectNamespace(); in groupOperationsByDialect()
|
| D | group_by_dialect.cc | 44 StringRef dialect = op.getName().getDialectNamespace(); in runOnOperation()
132 StringRef dialect = ops[0]->getName().getDialectNamespace(); in wrapOpsInFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/analysis/ |
| D | compatibility_analysis.cc | 64 if (op_name.getDialectNamespace() == in AnalyzeOperation()
65 mlir::TF::TensorFlowDialect::getDialectNamespace()) { in AnalyzeOperation()
75 } else if (op_name.getDialectNamespace() == "tf_executor") { in AnalyzeOperation()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | chlo_ops.h | 43 : Dialect(getDialectNamespace(), context, TypeID::get<ChloDialect>()) { in ChloDialect()
48 static StringRef getDialectNamespace() { return "chlo"; } in getDialectNamespace() function
|
| D | hlo_ops.h | 46 static StringRef getDialectNamespace() { return "mhlo"; } in getDialectNamespace() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/ |
| D | python_test_attrs.td | 33 return (getDialectNamespace() + ".static_type").str();
36 return (getDialectNamespace() + ".shape_value").str();
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.h | 42 static StringRef getDialectNamespace() { return "tf_device"; } in getDialectNamespace() function
|
| D | tf_executor.h | 38 static StringRef getDialectNamespace() { return "tf_executor"; } in getDialectNamespace() function
|
| D | tf_saved_model.h | 40 static StringRef getDialectNamespace() { return "tf_saved_model"; } in getDialectNamespace() function
|
| D | tf_dialect.h | 36 static StringRef getDialectNamespace() { return "tf"; } in getDialectNamespace() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/runtime_fallback/ |
| D | runtime_fallback_ops.h | 34 static StringRef getDialectNamespace() { return "tfd"; } in getDialectNamespace() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | xlir_ops.h | 41 static llvm::StringRef getDialectNamespace() { return "xlir"; } in getDialectNamespace() function
|
| /external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| D | tfrt_fallback_sync.h | 35 static StringRef getDialectNamespace() { return "tfrt_fallback_sync"; } in getDialectNamespace() function
|
| D | tfrt_fallback_async.h | 35 static StringRef getDialectNamespace() { return "tfrt_fallback_async"; } in getDialectNamespace() function
|
| D | tfrt_fallback.h | 33 static StringRef getDialectNamespace() { return "tfrt_fallback"; } in getDialectNamespace() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/ |
| D | tfrt_fallback_async.h | 35 static StringRef getDialectNamespace() { return "tfrt_fallback_async"; } in getDialectNamespace() function
|
| D | tfrt_fallback_sync.h | 35 static StringRef getDialectNamespace() { return "tfrt_fallback_sync"; } in getDialectNamespace() function
|
| D | tfrt_fallback.h | 33 static StringRef getDialectNamespace() { return "tfrt_fallback"; } in getDialectNamespace() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/opdefs/ |
| D | tf_cpurt_ops.h | 32 static mlir::StringRef getDialectNamespace() { return "tf_cpurt"; } in getDialectNamespace() function
|
| D | tf_jitrt_ops.h | 33 static mlir::StringRef getDialectNamespace() { return "tf_jitrt"; } in getDialectNamespace() function
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo/IR/ |
| D | lhlo_ops.h | 49 static StringRef getDialectNamespace() { return "lmhlo"; } in getDialectNamespace() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/ir/ |
| D | tfr_ops.h | 45 static StringRef getDialectNamespace() { return "tfr"; } in getDialectNamespace() function
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| D | ChloOps.h | 51 static StringRef getDialectNamespace() { return "chlo"; } in getDialectNamespace() function
|
| D | StablehloOps.h | 55 static StringRef getDialectNamespace() { return "stablehlo"; } in getDialectNamespace() function
|