Searched defs:tf_dialect (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | launch_to_device_attribute.cc | 39 LogicalResult AssignDevicesInRegion(const Dialect* tf_dialect, in AssignDevicesInRegion()
73 LogicalResult HoistOpsAndAnnotateWithDevice(const Dialect* tf_dialect, in HoistOpsAndAnnotateWithDevice()
95 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnOperation() local
|
| D | mark_ops_for_outside_compilation.cc | 232 const Dialect* tf_dialect) { in IsSupportedOp()
319 const Dialect* tf_dialect, Block* block, in MarkUncompilableOps()
350 bool ContainsUncompilableOps(const Dialect* tf_dialect, Block* block, in ContainsUncompilableOps()
393 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnOperation() local
|
| D | replicate_to_island.cc | 141 const Dialect* tf_dialect, OpBuilder& builder, in ExpandReplicateIntoReplicas()
236 LogicalResult CreateIslandsFromReplicate(const Dialect* tf_dialect, in CreateIslandsFromReplicate()
302 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnOperation() local
|
| D | device_attribute_to_launch.cc | 57 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnOperation() local
|
| D | replica_id_to_device_ordinal.cc | 50 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnOperation() local
|
| D | shape_inference.cc | 221 bool NeedsCastBack(OpOperand& use, Dialect* tf_dialect) { in NeedsCastBack()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 137 explicit Exporter(Graph* graph, const Dialect* tf_dialect) in Exporter()
424 const GraphExportConfig& configs, const Dialect* tf_dialect, in Convert()
589 const GraphExportConfig& configs, const Dialect* tf_dialect, in ConvertLibFunction()
675 auto tf_dialect = module.getContext()->getLoadedDialect("tf"); in Convert() local
774 Dialect* tf_dialect = func.getContext()->getLoadedDialect("tf"); in ConvertMlirFunctionToFunctionLibraryDef() local
|
| /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
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | layout_propagation_v2.cc | 69 mlir::ModuleOp* module, mlir::Dialect* tf_dialect, in AllOpResultsHaveLayouts()
139 mlir::ModuleOp module, mlir::Dialect* tf_dialect, in CopyLayoutsForSkippedOps()
1418 mlir::Dialect* tf_dialect = in runOnOperation() local
|
| D | mesh_propagation.cc | 448 mlir::Dialect* tf_dialect = in runOnOperation() local
|
| D | spmd_expander_common.cc | 430 mlir::ModuleOp* module, mlir::Dialect* tf_dialect, in PopulateConsumersFromModule()
|