- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for PartitionedCall (0.41 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
// CHECK-SAME: -> tensor<20xi32> func.func @stateful_partitioned_call(%arg0: tensor<20xi32>, %arg1: tensor<?xi32>) -> tensor<*xi32> { // CHECK: tf.PartitionedCall // CHECK-SAME: (tensor<20xi32>) -> tensor<20xi32> %0 = "tf.PartitionedCall"(%arg0) {config = "", config_proto = "", executor_type = "", f = @partitioned_called_func} : (tensor<20xi32>) -> tensor<*xi32> // CHECK: tf.StatefulPartitionedCall
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/lite/tests/prepare-composite-functions-tf.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 122.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
``` `_XlaMustCompile=true` in the following code ```mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
%3 = arith.constant dense<0> : tensor<i32> // CHECK: PartitionedCall // CHECK-SAME: f = @add // CHECK-SAME: device = "noodle" %4 = "tf.Case"(%2, %arg0, %arg1) {branches = [@sub, @add], output_shapes = [#tf_type.shape<>], device = "noodle", is_stateless = false} : (tensor<i32>, tensor<f32>, tensor<f32>) -> tensor<f32> // CHECK: PartitionedCall // CHECK-SAME: f = @sub // CHECK-SAME: _cluster_launch = "not_ready"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// `potential_refined_type`. Return true if the type was changed. bool RefineResultType(Operation* op, Value result, Type potential_refined_type); // Infers the shape from a (Stateful)PartitionedCall operation by looking up // the called function and propagating the return type. bool InferShapeForCall(CallOpInterface call_op); bool InferShapeForCast(Operation* op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K 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;
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/quantization/tensorflow/python/integration_test/quantize_model_test.py
# true branch doesn't. def _is_quantized_function_call_node( node_def: node_def_pb2.NodeDef, ) -> bool: return node_def.op == 'PartitionedCall' and node_def.attr[ 'f' ].func.name.startswith('quantized_') for func in output_graphdef.library.function: if func.signature.name.startswith('cond_false'):
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0)