- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 87 for statefulpartitionedcall (0.61 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir
// expected-remark@above {{ID: 0}} %r1 = "tf.ReadVariableOp"(%arg0) : (tensor<!tf_type.resource>) -> tensor<1xf32> // expected-remark@above {{ID: 1}} %two = "tf.StatefulPartitionedCall"(%one, %one) {config="", config_proto="", executor_type="", f=@add} : (tensor<1xf32>, tensor<1xf32>) -> tensor<1xf32> // expected-remark@above {{ID: 2}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 129.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
// (checked later). if (auto device_type_attr = op.getAttrOfType<StringAttr>(mlir::TF::kCompileDeviceTypeAttr)) { // tf.StatefulPartitionedCall ops with and without // _tpu_replicate attributes may exist in the same graph. Ops without // the attribute but with _XlaMustCompile=true would have // _xla_compile_device_type="" after
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
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
//===----------------------------------------------------------------------===// // StatefulPartitionedCall op legalization. //===----------------------------------------------------------------------===// // ----- // CHECK-LABEL: func @stateful_partitioned_call // CHECK-SAME: [[ARG:%.+]]: tensor<i32> func.func @stateful_partitioned_call(%arg0: tensor<i32>) -> tensor<i32> {
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 (2)