Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 11 for input_arg_size (0.17 sec)

  1. tensorflow/compiler/jit/get_compiler_ir.cc

      // `input_args` includes both concrete_fn input args and captured_input here.
      auto& input_args = fbody->record->fdef().signature().input_arg();
      // Here input_arg_size = len(flat_args) + len(captured_input)
      int input_arg_size = input_args.size();
    
      std::vector<const Tensor*> inputs(input_arg_size);
      std::deque<Tensor> inputs_storage;
      std::vector<VariableInfo> variable_infos;
      int offset = input_arg_shape_and_dtype.size();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc

      Builder builder(context_);
    
      if (current_ods_input_ != op_def_->input_arg_size())
        return InvalidArgument(absl::StrCat("Mismatch in operands number: got ",
                                            current_ods_input_, " expected ",
                                            op_def_->input_arg_size(), " ; for op ",
                                            state_->name.getStringRef().str()));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  3. tensorflow/cc/framework/fuzzing/cc_op_fuzz_gen.cc

        std::cout << "NOT fuzzing: " << op_info.graph_op_def.name()
                  << " has a constrained kernel.\n";
        return false;
      }
    
      for (int i = 0; i < op_info.graph_op_def.input_arg_size(); ++i) {
        const auto& arg(op_info.graph_op_def.input_arg(i));
        // TODO(unda, b/249298521): deal with inputs that are required to be refs
        if (arg.is_ref()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jan 27 16:26:51 UTC 2024
    - 13K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/saved_model/core/saved_model_utils.cc

      if (input_specs.size() + saved_concrete_function.bound_inputs_size() !=
          function_def->signature().input_arg_size()) {
        return errors::FailedPrecondition(
            "FunctionDef ", name, " has ",
            function_def->signature().input_arg_size(),
            " inputs, but the SavedConcreteFunction has ", input_specs.size(),
            " flattened user inputs and ",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jan 12 19:17:46 UTC 2023
    - 24K bytes
    - Viewed (0)
  5. tensorflow/c/ops_test.cc

      ::tensorflow::OpList op_list;
      op_list.ParseFromArray(op_list_buffer->data, op_list_buffer->length);
      bool found = false;
      for (const auto& op : op_list.op()) {
        if (op.name() == "SomeOp") {
          ASSERT_EQ(2, op.input_arg_size());
          ASSERT_EQ("input1", op.input_arg(0).name());
          ASSERT_EQ(::tensorflow::DT_UINT8, op.input_arg(0).type());
          ASSERT_EQ(1, op.attr_size());
          ASSERT_EQ("string", op.attr(0).type());
          found = true;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 11 01:20:50 UTC 2021
    - 12.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc

      // and output types should be {DT_BOOL}.
      const FunctionDef *f1_rewritten = fld.Find("f1_rearrange_0");
      CHECK_NE(f1_rewritten, nullptr);
      ASSERT_EQ(f1_rewritten->signature().input_arg_size(), 2);
      EXPECT_EQ(f1_rewritten->signature().input_arg(0).type(), DT_BOOL);
      EXPECT_EQ(f1_rewritten->signature().input_arg(1).type(), DT_RESOURCE);
      ASSERT_EQ(f1_rewritten->signature().output_arg_size(), 1);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 11:36:41 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  7. tensorflow/cc/framework/cc_op_gen.cc

      string body;
      string return_on_error =
          strings::StrCat("if (!", scope_str, ".ok()) return;");
    
      strings::StrAppend(&body, "  ", return_on_error, "\n");
    
      for (int i = 0; i < op_info.graph_op_def.input_arg_size(); ++i) {
        const auto& arg(op_info.graph_op_def.input_arg(i));
        const auto& api_def_arg(op_info.api_def.in_arg(i));
        strings::StrAppend(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 17:22:47 UTC 2023
    - 17K bytes
    - Viewed (0)
  8. tensorflow/c/c_api_function_test.cc

      }
    
      void VerifyFDefInputs(const tensorflow::FunctionDef& fdef,
                            const std::vector<IOSpec>& inputs) {
        const OpDef& signature = fdef.signature();
        ASSERT_EQ(inputs.size(), signature.input_arg_size());
        for (int i = 0; i < inputs.size(); ++i) {
          const OpDef::ArgDef& arg = signature.input_arg(i);
          const IOSpec& in = inputs[i];
          if (in.second != DT_INVALID) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 20 22:08:54 UTC 2023
    - 63.6K bytes
    - Viewed (0)
  9. tensorflow/cc/framework/cc_op_gen_util.cc

      }
    }
    
    void InferOpAttributes(
        const OpDef& op_def,
        std::unordered_map<string, string>* inferred_input_attrs) {
      for (int i = 0; i < op_def.input_arg_size(); ++i) {
        const auto& arg(op_def.input_arg(i));
        InferArgAttributes(arg, inferred_input_attrs);
      }
    }
    
    bool ArgIsList(const OpDef::ArgDef& arg) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 00:57:05 UTC 2024
    - 25K bytes
    - Viewed (0)
  10. tensorflow/c/c_api_experimental.cc

      const tensorflow::OpDef* op_def = nullptr;
      status->status =
          tensorflow::OpRegistry::Global()->LookUpOpDef(op_name, &op_def);
      if (!status->status.ok()) return nullptr;
    
      if (input_index >= op_def->input_arg_size() || input_index < 0) {
        status->status = tensorflow::errors::InvalidArgument(
            input_index, " out of range for ", op_name);
        return nullptr;
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 29.4K bytes
    - Viewed (0)
Back to top