- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for statefulpartitionedcall (0.76 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tensor_array_ops_decomposition.mlir
// CHECK: %[[GVAR2:.*]] = "tf.MlirLocalVarOp"() : () -> tensor<!tf_type.resource<tensor<5x3xf32>>> // CHECK: "tf.StatefulPartitionedCall"(%[[VAR]], %[[GVAR1]], %[[GVAR2]]) // CHECK-SAME: f = @callee_tensorarray_decomposed %call = "tf.StatefulPartitionedCall"(%ta#0) {f = @callee, config = "", config_proto = "", executor_type = ""} : (tensor<!tf_type.resource>) -> tensor<!tf_type.resource>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 49K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/resource_op_lifting.mlir
// CHECK: %[[CLUSTER:.*]] = "tf_device.cluster"() "tf_device.cluster"() ({ // CHECK: %[[PC0:.*]] = "tf.StatefulPartitionedCall"(%[[READ0]], %[[READ1]], %[[CONST]]) // CHECK-SAME: f = @callee_resource_lifted %3 = "tf.StatefulPartitionedCall"(%0, %1, %2) {f = @callee, config = "", config_proto = "", executor_type = ""}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 74K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
} } } } return LogicalResult::success(); } struct Callers { TF::StatefulPartitionedCallOp forward; TF::StatefulPartitionedCallOp core_tpu; TF::StatefulPartitionedCallOp backward; TF::StatefulPartitionedCallOp non_tpu; }; template <typename InputContainer> std::vector<Type> GetValueTypes(const InputContainer& input) {
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/jit/mark_for_compilation_pass.cc
"SpaceToBatchND", "SpaceToDepth", "SparseMatMul", "SparseToDense", "StackCloseV2", "StackPopV2", "StackPushV2", "StackV2", "StatefulPartitionedCall", "StatefulStandardNormalV2", "StatefulTruncatedNormal", "StatefulUniform", "StatefulUniformFullInt", "StatefulUniformInt", "StatelessCase",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K 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) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
callee.getContext(), callee.getFunctionType().getInputs(), llvm::to_vector<4>(new_return.getOperandTypes()))); return success(); } // Updates a PartitionedCallOp/StatefulPartitionedCallOp according to the // resource-lifted new callee function in lifting_info. template <typename CallOpType> void UpdatePartitionedCallOpWithNewCallee(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0)