- Sort Score
- Result 10 results
- Languages All
Results 21 - 25 of 25 for StatefulPartitionedCall (0.3 sec)
-
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/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/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/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)