- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for StatefulPartitionedCall (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
// TPUCompilationResultsOp. if (op->hasAttr(kTpuCompilationStatus)) { op->setAttr(kTpuCompilationStatus, attr); } } // A helper class to inline TF::StatefulPartitionedCall ops struct Inliner : public InlinerInterface { Inliner(OpBuilder& builder, SymbolTable& symbol_table) : InlinerInterface(builder.getContext()), builder(builder),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
TF_Tensor:$output ); TF_DerivedResultTypeAttr dtype = TF_DerivedResultTypeAttr<0>; DerivedAttr shape = TF_DerivedResultShapeAttr; } def TF_StatefulPartitionedCallOp : TF_Op<"StatefulPartitionedCall", [CallOpInterface, DeclareOpInterfaceMethods<SymbolUserOpInterface>]> { let summary = "returns `f(inputs)`, where `f`'s body is placed and partitioned."; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0)