- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for _disable_call_shape_inference (0.31 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/convert_to_legacy_compile_and_replicate_attributes.mlir
// CHECK-LABEL: func @convert_to_legacy_attribute func.func @convert_to_legacy_attribute(%arg0: tensor<*xf32>, %arg1: tensor<f32>, %arg2: tensor<f32>) -> tensor<*xf32> attributes {tf._disable_call_shape_inference = true, tf.signature.is_stateful} { // CHECK-NOT: _replication_info = %0 = tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
op->getSuccessors(), new_regions); inner_op = rewriter.create(state); } else { bool disable_call_shape_inference = false; if (op->hasAttr("_disable_call_shape_inference")) { disable_call_shape_inference = op->getAttrOfType<BoolAttr>("_disable_call_shape_inference") .getValue(); } inner_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td
$config, $config_proto, $executor_type), (CallOp $f, $args), [(ArgTypesMatchCallee $op, $args, $f)]>; } // The extra attr on this op is _disable_call_shape_inference, which we ignore // in the bridge. def : Pat<(TF_LegacyCallOp:$op $args, FlatSymbolRefAttr:$f, $attr), (CallOp $f, $args), [(ArgTypesMatchCallee $op, $args, $f)]>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 34.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
mlir::SymbolRefAttr val = mlir::SymbolRefAttr::get(builder_.getContext(), node_type_name); result.addAttribute("f", val); if (!result.attributes.get("_disable_call_shape_inference")) { result.addAttribute("_disable_call_shape_inference", builder_.getBoolAttr(false)); } } auto composite_control_flow_op = [&](const std::string& name) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
instead executes it on the device assigned to this op. }]; let arguments = (ins Variadic<TF_Tensor>:$args, FlatSymbolRefAttr:$f, DefaultValuedOptionalAttr<BoolAttr, "false">:$_disable_call_shape_inference ); let results = (outs Variadic<TF_Tensor>:$output ); let extraClassDeclaration = [{ // Gets the argument operands to the called function.
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/tf2xla/tests/legalize-tf.mlir
// CHECK: %[[RESULT:.*]] = call @select_first(%arg0, %arg1) : (tensor<10x2xf32>, tensor<10x2xf32>) -> tensor<10x2xf32> %0 = "tf.LegacyCall"(%arg0, %arg1) {_disable_call_shape_inference = true, _tpu_replicate = "cluster", device = "", f = @select_first} : (tensor<10x2xf32>, tensor<10x2xf32>) -> tensor<10x2xf32> // CHECK: return %[[RESULT]] func.return %0: tensor<10x2xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0)