Searched defs:tf_dialect (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | replicate_to_island.cc | 50 const Dialect* tf_dialect, OpBuilder* builder, in ExpandReplicateIntoReplicas()
140 void CreateIslandsFromReplicate(const Dialect* tf_dialect, in CreateIslandsFromReplicate()
189 void LowerSingleIslandReplicateToIslands(const Dialect* tf_dialect, in LowerSingleIslandReplicateToIslands()
199 const Dialect* tf_dialect = getContext().getRegisteredDialect("tf"); in runOnFunction() local
|
| D | shape_inference.cc | 112 Dialect* tf_dialect, Type old_type) { in AddCastBackForUnsupportedNonTFUses()
146 Operation* op, Dialect* tf_dialect) { in InferShapeForPassThroughOps()
161 bool InferShapeForNonTFDialectOperation(Operation* op, Dialect* tf_dialect) { in InferShapeForNonTFDialectOperation()
182 bool InferShapeForSingleOperation(Operation* op, Dialect* tf_dialect, in InferShapeForSingleOperation()
462 Dialect* tf_dialect = ctx->getRegisteredDialect<TensorFlowDialect>(); in InferShapeUntilFixPoint() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 206 explicit Exporter(Graph* graph, const Dialect* tf_dialect) in Exporter()
519 const GraphExportConfig& configs, const Dialect* tf_dialect, in Convert()
717 const Dialect* tf_dialect, in ConvertLibFunction()
800 auto tf_dialect = module.getContext()->getRegisteredDialect("tf"); in Convert() local
|