- Sort Score
- Result 10 results
- Languages All
Results 61 - 68 of 68 for StatefulPartitionedCall (1 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu_sharding_identification.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 20 19:07:52 UTC 2024 - 47.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
func.return %arg0 : tensor<i32> } // ----- // Test valid tf.StatefulPartitionedCall // CHECK-LABEL: func @testValidStatefulPartitionedCall func.func @testValidStatefulPartitionedCall(%arg0: tensor<i32>) -> tensor<i32> { %0 = "tf.StatefulPartitionedCall"(%arg0) {config = "", config_proto = "", executor_type = "", f = @pcall_func} : (tensor<i32>) -> (tensor<i32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK: call @callee() : () -> () // CHECK: call @callee() : () -> () "tf.PartitionedCall"() {config = "", config_proto = "", executor_type = "", f = @callee} : () -> () "tf.StatefulPartitionedCall"() {config = "", config_proto = "", executor_type = "", f = @callee} : () -> () "tf.LegacyCall"() {config = "", config_proto = "", executor_type = "", f = @callee} : () -> () func.return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K 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/cc/saved_model/testdata/chunked_saved_model/chunked_model/saved_model.pbtxt
} } attr { key: "shape" value { shape { } } } } node { name: "StatefulPartitionedCall" op: "StatefulPartitionedCall" input: "saver_filename" input: "Const_1" attr { key: "Tin" value { list { type: DT_STRING type: DT_STRING
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 21:43:11 UTC 2023 - 531.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
return false; } // Returns whether the current op is not supported by the TF Lite runtime. static bool IsUnsupportedFlexOp(const std::string& op_name) { return op_name == "PartitionedCall" || op_name == "StatefulPartitionedCall"; } // Create description of operation that could not be converted. static std::string GetOpDescriptionForDebug(Operation* inst) { const int kLargeElementsAttr = 16; std::string op_str;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
%2 = "tf.If"(%0, %arg0, %arg1) {then_branch = @add, else_branch = @sub, output_shapes = [#tf_type.shape<>], device = "noodle", is_stateless = true} : (tensor<i1>, tensor<f32>, tensor<f32>) -> tensor<f32> // CHECK: %1 = "tf.StatefulPartitionedCall"(%0, %arg1) // CHECK-SAME: f = @add // CHECK-SAME: _underscore_attr = "something" // CHECK-SAME: device = "noodle"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0)