- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for LegacyCallOp (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/export_utils.h
// Returns true if the given instruction is an mlir::TF::LegacyCallOp or the // result of such an operation transformed by the // ExecutorToControlDialectConversion pass. // // TODO(b/145706023): When the ExecutorToControlDialectConversion pass runs // before the exporter, it mutates an mlir::TF::LegacyCallOp instruction to // an instruction with a different operation name. As such, this routine checks
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir_wrapper/ops.cc
}); // mlir::TF::LegacyCallOp py::class_<mlir::TF::LegacyCallOp>(m, "Tf_LegacyCallOp") .def("create", [](mlir::OpBuilder& opb, mlir::Location loc, std::vector<mlir::Type> output, std::vector<mlir::Value> args, std::string f) -> mlir::Operation* { return opb .create<mlir::TF::LegacyCallOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 30 02:12:49 UTC 2022 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
bool SupportsCommunicationComputation(Operation* op) { return isa<TF::IfRegionOp, TF::WhileRegionOp, TF::CaseRegionOp, TF::XlaCallModuleOp, TF::StatefulPartitionedCallOp, TF::PartitionedCallOp, TF::LegacyCallOp>(op); } #define GEN_PASS_DEF_PREPARETPUCOMPUTATIONFORTFEXPORTPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" class PrepareTpuComputationForTfExportPass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
disable_call_shape_inference = op->getAttrOfType<BoolAttr>("_disable_call_shape_inference") .getValue(); } inner_op = rewriter.create<LegacyCallOp>(loc, new_types, inner_op_operands, op_name, disable_call_shape_inference); } rewriter.create<tf_executor::YieldOp>(loc, inner_op->getResults());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
actual_shape.ShortDebugString()); } } return absl::OkStatus(); } bool IsLegacyCallInstruction(mlir::Operation* inst) { return llvm::dyn_cast<mlir::TF::LegacyCallOp>(inst); } Status AddTensorFlowOpPrefix(std::string prefix) { GlobalOpPrefixes()->insert(prefix); return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::IdentityNOp>(), TypeID::get<TF::InplaceUpdateOp>(), TypeID::get<TF::InvertPermutationOp>(), TypeID::get<TF::IRFFTOp>(), TypeID::get<TF::L2LossOp>(), TypeID::get<TF::LegacyCallOp>(), TypeID::get<TF::LessEqualOp>(), TypeID::get<TF::LinSpaceOp>(), TypeID::get<TF::LogicalOrOp>(), TypeID::get<TF::MaxOp>(), TypeID::get<TF::MaximumOp>(), TypeID::get<TF::MaxPoolOp>(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
} return {}; } //===----------------------------------------------------------------------===// // LegacyCallOp //===----------------------------------------------------------------------===// LogicalResult LegacyCallOp::verifySymbolUses( SymbolTableCollection& symbolTable) { StringAttr func_attr = getFAttr().getAttr(); StringRef func_name = func_attr.getValue();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0)