Searched refs:tf_dialect_ (Results 1 – 3 of 3) sorted by relevance
557 Dialect* const tf_dialect_; member in mlir::TF::ShapeInference619 : tf_dialect_(context->getLoadedDialect<TensorFlowDialect>()), in ShapeInference()662 else if (NeedsCastBack(use, tf_dialect_)) in UpdateTypeAndInsertIncompatibleUseCasts()716 return NeedsCastBack(use, tf_dialect_); in InferShapeForCast()1035 assert(tf_dialect_ == op->getDialect()); in InferShapeForSingleOperation()1421 if (some_unknown && dialect == tf_dialect_) return failure(); in TryToFold()1544 if (op->getDialect() != tf_dialect_) { in InferShapeUntilFixPoint()
177 : graph_(graph), tf_dialect_(tf_dialect) {} in Exporter()207 const mlir::Dialect* tf_dialect_; member in tensorflow::__anon29e43afe0111::Exporter
478 tf_dialect_ = in Translator()627 const Dialect* tf_dialect_; member in __anon4d1313dc0111::Translator1167 if (dialect == tf_dialect_) { in BuildOperator()