- Sort Score
- Result 10 results
- Languages All
Results 81 - 87 of 87 for statefulpartitionedcall (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_validate_inputs.cc
} bool IsPartitionedOp(Operation* op) { static auto* ops = [] { llvm::SmallDenseSet<mlir::TypeID, 32>* ops_set = new llvm::SmallDenseSet<mlir::TypeID, 32>{ TypeID::get<TF::StatefulPartitionedCallOp>(), TypeID::get<TF::PartitionedCallOp>(), TypeID::get<TF::TPUPartitionedCallOp>(), }; return ops_set; }(); auto abstractOp = op->getRegisteredInfo();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 06:51:01 UTC 2024 - 21.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::SqrtGradOp>(), TypeID::get<TF::SquaredDifferenceOp>(), TypeID::get<TF::SqueezeOp>(), TypeID::get<TF::StatelessParameterizedTruncatedNormalOp>(), TypeID::get<TF::StatefulPartitionedCallOp>(), TypeID::get<TF::StopGradientOp>(), TypeID::get<TF::StridedSliceOp>(), TypeID::get<TF::StridedSliceGradOp>(), TypeID::get<TF::SumOp>(), TypeID::get<TF::TanhGradOp>(),
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/transforms/passes.h
// tf_device.launch op no longer exist after Outside Compilation is complete. std::unique_ptr<OperationPass<func::FuncOp>> CreateVerifyNoOutsideCompilationMarkersPass(); // Create a pass that inlines the StatefulPartitionedCallOp op based in the // parent region. std::unique_ptr<OperationPass<ModuleOp>> CreateXlaInlineDeviceOpsPass(); // Creates a pass that rewrites partitioned calls with `_xla_compile_device
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
bool cond = cond_attr.getSplatValue<BoolAttr>().getValue(); FlatSymbolRefAttr func = cond ? op.getThenBranchAttr() : op.getElseBranchAttr(); // Replace IfOp with PartitionedCallOp or StatefulPartitionedCallOp. auto rewrite = [&](auto op_type) { auto empty = rewriter.getStringAttr(""); ReplaceTfOpWithNewOp<typename decltype(op_type)::CallOp>( rewriter, op, op.getResultTypes(), op.getInput(), func,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
} return success(); } LogicalResult PartitionedCallOp::verifySymbolUses( SymbolTableCollection &symbolTable) { return VerifyPartitionedCall(*this, symbolTable); } LogicalResult StatefulPartitionedCallOp::verifySymbolUses( SymbolTableCollection &symbolTable) { return VerifyPartitionedCall(*this, symbolTable); } LogicalResult TPUPartitionedCallOp::verifySymbolUses( SymbolTableCollection &symbolTable) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
args_as_values.push_back(block_argument); } mlir::OpBuilder body_builder(&func.getBody()); auto call = body_builder.create<mlir::TF::StatefulPartitionedCallOp>( func.getLoc(), orig_func.getFunctionType().getResults(), args_as_values, mlir::SymbolRefAttr::get(builder.getContext(), orig_func.getName()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)