Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 12 of 12 for num_inputs (0.16 sec)

  1. tensorflow/compiler/jit/variable_info_util.cc

      }
      return absl::OkStatus();
    }
    
    std::vector<int> GetResourceVariableIndicesFromContext(OpKernelContext* ctx) {
      std::vector<int> out;
      for (int64 i = 0; i < ctx->num_inputs(); i++) {
        if (ctx->input(i).dtype() == DT_RESOURCE) {
          out.push_back(i);
        }
      }
      return out;
    }
    
    Status CreateVariableInfoLookup(
        absl::Span<VariableInfo const> variable_args,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 7K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/ops/gen/cpp/views/op_view.cc

    const std::vector<AttrView>& OpView::Attributes() const {
      return argument_attrs_;
    }
    
    const std::vector<OpArgumentView>& OpView::AllArguments() const {
      return all_arguments_;
    }
    
    int OpView::NumInputs() const { return input_args_.size(); }
    
    int OpView::NumOutputs() const { return output_args_.size(); }
    
    ArgView OpView::OnlyInput() const {
      CHECK_EQ(input_args_.size(), 1);  // Crash OK
      return input_args_.front();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 03 07:02:00 UTC 2024
    - 3.2K bytes
    - Viewed (0)
Back to top