- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 35 for PartitionedCall (1.24 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantized_function_library_xla_weight_only.mlir
attributes {tf_quant.quantized_ops = ${quantized_ops}} { %accum_out = "tf.PartitionedCall"(%input, %weight) { config = "", config_proto = "", executor_type = "", f=@${internal_func_name} } : (tensor<*xf32>, tensor<*xi8>) -> tensor<*xf32> %out = "tf.PartitionedCall"(%accum_out, %weight_scale) { config = "", config_proto = "", executor_type = "", f=@internal_dequantize_f32
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 15:43:38 UTC 2023 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/fake_quant_e2e_xla.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_to_hlo_pipeline/sccp-post-shape-inference.mlir
%1 = "tf.Reshape"(%arg1, %0) : (tensor<19x10xf32>, tensor<2xi64>) -> tensor<?x?xf32> // CHECK: %[[RESULT:.*]] = mhlo.constant dense<[10, 19]> %2 = "tf.PartitionedCall"(%1) {config = "", config_proto = "", executor_type = "", f = @get_shape} : (tensor<?x?xf32>) -> (tensor<?xi64>) // CHECK: return %[[RESULT]] func.return %2 : tensor<?xi64> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jul 25 02:54:34 UTC 2023 - 1020 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/prepare_tpu_computation_for_tf_export.mlir
// CHECK: _xla_original_oc_node_name = [[NODE_NAME1:.*]], _xla_token_input_nodes = ["_xla_token_arg_node"] %0 = "tf.PartitionedCall"(%arg0) {config = "", config_proto = "", executor_type = "", f = @PartitionedCall3} : (tensor<i32>) -> (tensor<i32>) // CHECK-NOT: _xla_token_input_nodes %1 = "tf.PartitionedCall"(%0) {config = "", config_proto = "", executor_type = "", f = @IdentityFunc} : (tensor<i32>) -> (tensor<i32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 18:46:36 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/insert_calibration_statistics_saver_with_skipping.mlir
func.func @serving_default(%arg0: tensor<1x3x4x3xf32>) -> (tensor<1x2x2x2xf32>) attributes {tf.entry_function = {control_outputs = "", inputs = "serving_default_input_tensor:0", outputs = "PartitionedCall:0"}} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
Option<bool> skip_partitioned_calls_{ *this, "skip-partitioned-calls", ::llvm::cl::desc( "Skip tf.StatefulPartitionedCall and tf.PartitionedCall")}; }; void TFToMhloPass::runOnOperation() { auto func = getOperation(); MLIRContext *context = func->getContext(); RewritePatternSet patterns(context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/remove_unused_arguments.mlir
return %arg0 : tensor<f32> } // CHECK-LABEL: handles_partitioned_function_calls func.func @handles_partitioned_function_calls(%arg0: tensor<f32>, %arg1: tensor<f32>) -> tensor<f32> { // CHECK: PartitionedCall"() %1 = "tf.PartitionedCall"(%arg0, %arg1) {f = @f} : (tensor<f32>, tensor<f32>) -> tensor<f32> return %1 : tensor<f32> } // ----- func.func private @f(%arg0: tensor<f32>, %arg1: tensor<f32>) -> tensor<f32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 06 23:00:44 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/jit/force_xla_constants_on_host_pass_test.cc
Output in = ops::Placeholder(root, DT_FLOAT); Output perm = ops::Const(root, {3, 1, 2, 0}); NameAttrList b_name_attr; b_name_attr.set_name("TransposeCall"); ops::PartitionedCall call(root.WithOpName("call"), {in, perm}, {DT_FLOAT}, b_name_attr); call.output.front().node()->AddAttr(kXlaMustCompileAttr, true); std::unique_ptr<Graph> graph;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.cc
auto value = device_attr.getValue(); // TODO(b/229028654): Remove string conversion once we have C++17. absl::string_view device_type(value.data(), value.size()); // Device type may be empty for some ops, e.g. tf.PartitionedCall. auto it = std::find(kValidDeviceTypes.begin(), kValidDeviceTypes.end(), device_type); if (it == kValidDeviceTypes.end()) return failure(); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-as-function.pbtxt
} } } attr { key: "Tout" value { list { type: DT_FLOAT } } } attr { key: "_gradient_op_type" value { s: "PartitionedCall-1205" } } attr { key: "config" value { s: "" } } attr { key: "config_proto" value {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 24 00:18:34 UTC 2023 - 5K bytes - Viewed (0)