Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 28 for num_inputs (0.25 sec)

  1. tensorflow/c/c_api_experimental.cc

      using tensorflow::shape_inference::ShapeHandle;
    
      const int num_inputs = input_shapes->num_items;
      NodeDef node_def;
      tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(tfe_op);
      node_def.set_name(op->Name());
      node_def.set_op(op->Name());
      for (int i = 0; i < num_inputs; ++i) {
        node_def.add_input("dummy_input");
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 29.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc

        if (data_type.getIntOrFloatBitWidth() == 64) continue;
    
        const auto& block_arg = replicate.GetBody().getArgument(replicate_arg);
    
        int64_t num_inputs = 0;
        if (replicate.IsReplicatedBlockArgument(block_arg)) {
          num_inputs = num_replicas;
        } else {
          num_inputs = 1;
        }
    
        // We have found a mirrored variable which is an input to the replicated
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_launch_util.cc

      return variable_lookup;
    }
    
    }  // anonymous namespace
    
    std::vector<const Tensor*> InputsFromContext(OpKernelContext* ctx) {
      std::vector<const Tensor*> inputs;
      inputs.reserve(ctx->num_inputs());
      for (int input_idx = 0; input_idx < ctx->num_inputs(); input_idx++) {
        inputs.push_back(&ctx->input(input_idx));
      }
      return inputs;
    }
    
    absl::StatusOr<std::vector<int>> GetConstantInputIndicesFromContext(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 00:36:08 UTC 2024
    - 40.4K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_unified_experimental.cc

                   absl::StatusMessageAsCStr(status));
    }
    
    void TF_ExecuteOperation(TF_AbstractOp* op, int num_inputs,
                             TF_AbstractTensor* const* inputs, TF_OutputList* o,
                             TF_Status* s) {
      for (int i = 0; i < num_inputs; i++) {
        tsl::Set_TF_Status_from_Status(s, unwrap(op)->AddInput(unwrap(inputs[i])));
        if (TF_GetCode(s) != TF_OK) {
          return;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 10:15:17 UTC 2024
    - 9K bytes
    - Viewed (0)
  5. tensorflow/cc/framework/ops.h

    /// @addtogroup core
    /// @{
    
    /// Represents a node in the computation graph.
    class Operation {
     public:
      Operation() : node_(nullptr) {}
      explicit Operation(Node* n);
    
      int32 num_inputs() const { return node_->num_inputs(); }
      DataType input_type(int32_t o) const { return node_->input_type(o); }
      Output input(int32_t i) const;
    
      int32 num_outputs() const { return node_->num_outputs(); }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  6. tensorflow/c/kernels.cc

      }
    #endif  // defined(IS_MOBILE_PLATFORM) || defined(IS_SLIM_BUILD)
    }
    
    int TF_NumInputs(TF_OpKernelContext* ctx) {
      auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx);
      return cc_ctx->num_inputs();
    }
    
    int TF_NumOutputs(TF_OpKernelContext* ctx) {
      auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx);
      return cc_ctx->num_outputs();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 36K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op.cc

            absl::AbortedError(
                "The `ids` and `calibration_methods` must have the same size."));
    
        // Check the number and type of inputs.
        OP_REQUIRES(context, context->num_inputs() == ids_.size() * 3,
                    absl::AbortedError("The number of inputs must be  three times "
                                       "the size of the `ids` list."));
        for (int i = 0; i < ids_.size(); ++i) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 13 01:31:23 UTC 2024
    - 8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc

        auto input = pos_and_input.value();
        bool is_packed = input.getIsPacked();
        const int num_operands = input->getNumOperands();
        int num_inputs = is_packed ? 1 : num_replicas;
        if (num_operands != num_inputs)
          return input->emitOpError() << "requires " << num_inputs << " operands";
        if (is_packed) {
          packed_inputs.push_back(input->getOperand(0));
          packed_ops.push_back(input);
        } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 39.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/shape_inference.cc

                                     switch_input == n;
            if (is_loop_invariant) {
              shape_inference::InferenceContext* context =
                  shape_refiner->GetContext(n);
              for (int i = 0; i < n->num_inputs(); i++) {
                const Node* input_node;
                if (n->input_node(i, &input_node).ok()) {
                  auto shapes_and_types = context->input_handle_shapes_and_types(i);
                  if (shapes_and_types) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 00:41:19 UTC 2024
    - 13K bytes
    - Viewed (0)
  10. tensorflow/c/c_api.cc

    }
    
    void TF_AddInputList(TF_OperationDescription* desc, const TF_Output* inputs,
                         int num_inputs) {
      std::vector<NodeBuilder::NodeOut> input_list;
      input_list.reserve(num_inputs);
      for (int i = 0; i < num_inputs; ++i) {
        input_list.emplace_back(&inputs[i].oper->node, inputs[i].index);
      }
      desc->node_builder.Input(input_list);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
Back to top