Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for input_arg (0.7 sec)

  1. tensorflow/c/c_api_experimental.cc

      }
    
      const tensorflow::OpDef_ArgDef& input_arg = op_def->input_arg()[input_index];
    
      if (input_arg.number_attr().empty()) {
        status->status = tensorflow::errors::NotFound(
            op_name, " does not have number_attr() defined.");
        return nullptr;
      }
    
      // The returned string is owned by OpRegistry, so liveness is not a concern.
      return input_arg.number_attr().c_str();
    }
    
    Registered: Tue Sep 09 12:39:10 UTC 2025
    - Last Modified: Mon Aug 18 03:53:25 UTC 2025
    - 29.5K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_function_test.cc

        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) {
            ASSERT_EQ(arg.type(), in.second)
                << "Got unexpected type for input " << i
                << ". fdef: " << fdef.DebugString();
    Registered: Tue Sep 09 12:39:10 UTC 2025
    - Last Modified: Thu Jul 20 22:08:54 UTC 2023
    - 63.6K bytes
    - Viewed (1)
Back to top