| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | launch_to_device_attribute.cc | 39 LogicalResult AssignDevicesInRegion(const Dialect* tf_dialect, in AssignDevicesInRegion() argument
43 if (op->getDialect() != tf_dialect) return WalkResult::advance(); in AssignDevicesInRegion()
73 LogicalResult HoistOpsAndAnnotateWithDevice(const Dialect* tf_dialect, in HoistOpsAndAnnotateWithDevice() argument
79 if (failed(AssignDevicesInRegion(tf_dialect, launch, launch.body()))) in HoistOpsAndAnnotateWithDevice()
95 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnOperation() local
96 if (!tf_dialect) { in runOnOperation()
101 auto result = getOperation().walk([&tf_dialect](tf_device::LaunchOp launch) { in runOnOperation()
102 if (failed(HoistOpsAndAnnotateWithDevice(tf_dialect, launch))) in runOnOperation()
|
| D | mark_ops_for_outside_compilation.cc | 232 const Dialect* tf_dialect) { in IsSupportedOp() argument
233 if (op.getDialect() != tf_dialect) in IsSupportedOp()
319 const Dialect* tf_dialect, Block* block, in MarkUncompilableOps() argument
330 if (!IsSupportedOp(*op, supported_ops, tf_dialect)) { in MarkUncompilableOps()
350 bool ContainsUncompilableOps(const Dialect* tf_dialect, Block* block, in ContainsUncompilableOps() argument
363 if (!IsSupportedOp(*op, supported_ops, tf_dialect)) { in ContainsUncompilableOps()
393 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnOperation() local
394 if (!tf_dialect) { in runOnOperation()
426 if (failed(MarkUncompilableOps(tf_dialect, &cluster.GetBody(), in runOnOperation()
430 if (ContainsUncompilableOps(tf_dialect, &cluster.GetBody(), in runOnOperation()
|
| D | replicate_to_island.cc | 141 const Dialect* tf_dialect, OpBuilder& builder, in ExpandReplicateIntoReplicas() argument
236 LogicalResult CreateIslandsFromReplicate(const Dialect* tf_dialect, in CreateIslandsFromReplicate() argument
245 if (failed(ExpandReplicateIntoReplicas(tf_dialect, builder, island_op, in CreateIslandsFromReplicate()
302 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnOperation() local
303 if (!tf_dialect) { in runOnOperation()
324 if (failed(CreateIslandsFromReplicate(tf_dialect, graph_op, island_op, in runOnOperation()
|
| D | device_attribute_to_launch.cc | 57 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnOperation() local
60 if (op->getDialect() != tf_dialect) return WalkResult::advance(); in runOnOperation()
|
| D | replica_id_to_device_ordinal.cc | 50 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnOperation() local
51 if (!tf_dialect) { in runOnOperation()
|
| D | shape_inference.cc | 221 bool NeedsCastBack(OpOperand& use, Dialect* tf_dialect) { in NeedsCastBack() argument
222 return use.getOwner()->getDialect() != tf_dialect && in NeedsCastBack()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 122 const GraphExportConfig& configs, const Dialect* tf_dialect,
131 const GraphExportConfig& configs, const Dialect* tf_dialect,
137 explicit Exporter(Graph* graph, const Dialect* tf_dialect) in Exporter() argument
138 : graph_(graph), tf_dialect_(tf_dialect) {} in Exporter()
424 const GraphExportConfig& configs, const Dialect* tf_dialect, in Convert() argument
456 Exporter exporter(graph.get(), tf_dialect); in Convert()
523 TF_RETURN_IF_ERROR(ConvertLibFunction(configs, tf_dialect, symbol_table, in Convert()
589 const GraphExportConfig& configs, const Dialect* tf_dialect, in ConvertLibFunction() argument
602 Exporter::Convert(configs, tf_dialect, symbol_table, function, flib, in ConvertLibFunction()
628 TF_RETURN_IF_ERROR(ConvertLibFunction(configs, tf_dialect, symbol_table, in ConvertLibFunction()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/tensorflow/ |
| D | fallback_to_flex_ops.cc | 283 auto tf_dialect = ctx->getLoadedDialect<TF::TensorFlowDialect>(); in runOnOperation() local
285 if (op->getDialect() != tf_dialect) return; in runOnOperation()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | layout_propagation_v2.cc | 69 mlir::ModuleOp* module, mlir::Dialect* tf_dialect, in AllOpResultsHaveLayouts() argument
72 if (op->getDialect() != tf_dialect || in AllOpResultsHaveLayouts()
139 mlir::ModuleOp module, mlir::Dialect* tf_dialect, in CopyLayoutsForSkippedOps() argument
1418 mlir::Dialect* tf_dialect = in runOnOperation() local
1444 PopulateConsumersFromModule(&module, tf_dialect, consumers))) in runOnOperation()
1551 CopyLayoutsForSkippedOps(module, tf_dialect, merged_layouts))) in runOnOperation()
1558 if (!AllOpResultsHaveLayouts(&module, tf_dialect, merged_layouts)) in runOnOperation()
|
| D | spmd_expander_common.h | 106 mlir::ModuleOp* module, mlir::Dialect* tf_dialect,
|
| D | mesh_propagation.cc | 448 mlir::Dialect* tf_dialect = in runOnOperation() local
463 PopulateConsumersFromModule(&module, tf_dialect, consumers))) in runOnOperation()
|
| D | spmd_expander_common.cc | 430 mlir::ModuleOp* module, mlir::Dialect* tf_dialect, in PopulateConsumersFromModule() argument
440 if (op->getDialect() != tf_dialect) return; in PopulateConsumersFromModule()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ |
| D | BUILD | 396 "ir/tf_dialect.h",
431 "ir/tf_dialect.h",
481 "ir/tf_dialect.h",
526 "ir/tf_dialect.h",
571 "ir/tf_dialect.h",
644 "ir/tf_dialect.h",
671 "ir/tf_dialect.h",
|