Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for OpProducesOrConsumesVariant (0.23 sec)

  1. tensorflow/compiler/jit/compilability_check_util.h

               op_name == "RandomUniformInt" || op_name == "RandomStandardNormal" ||
               op_name == "TruncatedNormal" || op_name == "Multinomial";
      }
    
      bool OpProducesOrConsumesVariant(const Node& node) const {
        auto is_variant = [](DataType dtype) { return dtype == DT_VARIANT; };
        return absl::c_any_of(node.input_types(), is_variant) ||
               absl::c_any_of(node.output_types(), is_variant);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 14.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/compilability_check_util.cc

        LogNotCompilable(node, uncompilable_reason);
        return false;
      }
    
      if (!op_filter_.allow_ops_producing_or_consuming_variant &&
          OpProducesOrConsumesVariant(node)) {
        absl::string_view uncompilable_reason = "DT_VARIANT producer/consumer";
        MaybeMarkUncompilableNode(uncompilable_reason, *stack_trace,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 30.3K bytes
    - Viewed (0)
Back to top