Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for input_tensors (0.18 sec)

  1. tensorflow/c/c_api_experimental.cc

        // make the pointers invalid.
        all_input_tensors.reserve(num_inputs);
        for (int i = 0; i < num_inputs; ++i) {
          if (input_tensors[i] == nullptr) continue;
          all_input_tensors.emplace_back();
          Tensor& input_tensor = all_input_tensors.back();
          status->status = TF_TensorToTensor(input_tensors[i], &input_tensor);
          if (!status->status.ok()) return;
          input_tensors_vector[i] = &input_tensor;
        }
      }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_function.cc

    // does various checks while doing so. `input_nodes` will contain the same
    // information as input_tensors just in a different structure to make
    // following processing easier. TODO(iga): Simplify this nested structure.
    Status ProcessInputs(
        const TF_Graph* fn_body, const char* fn_name, int ninputs,
        const TF_Output* inputs, std::vector<OutputTensor>* input_tensors,
        std::unordered_map<const Node*, std::vector<int>>* input_nodes)
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 13.6K bytes
    - Viewed (2)
  3. tensorflow/c/c_api_experimental_test.cc

          }
        }
        TF_ShapeAndTypeList* output_shapes;
        TFE_InferShapes(op, input_shapes,
                        input_tensors.empty()
                            ? nullptr
                            : const_cast<TF_Tensor**>(input_tensors.data()),
                        /*input_tensors_as_shapes*/ nullptr,
                        /*input_resource_shapes_and_types*/ nullptr, &output_shapes,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Jan 17 22:27:52 GMT 2023
    - 13.1K bytes
    - Viewed (1)
  4. tensorflow/c/eager/tape.h

      });
      std::vector<Gradient*> in_grads;
      in_grads.reserve(input_tensors.size());
      for (int target_index = 0; target_index < input_tensors.size();
           ++target_index) {
        const auto current_grad =
            accumulated_gradients_.find(input_tensors[target_index].GetID());
        if (current_grad == accumulated_gradients_.end()) {
          if (IsDtypeTrainable(input_tensors[target_index].GetDType())) {
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  5. tensorflow/c/c_api_experimental.h

    //     OK to not have the inputs properly set in `op`. See `input_tensors`
    //     if you want shape inference to consider the input tensors of the
    //     op for shape inference.
    //   - The types need not be set in `input_shapes` as it is not used.
    //   - The number of `input_tensors` should be the same as the number of items
    //     in `input_shapes`.
    //
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 15.1K bytes
    - Viewed (0)
Back to top