Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for _disable_call_shape_inference (0.2 sec)

  1. 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)
  2. 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)
Back to top