- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for _disable_call_shape_inference (0.35 sec)
-
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)