Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 91 for tensors_ (0.16 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.h

          TF_Status* status);
    
      size_t num_tensors() const { return tensors_.size(); }
      TFE_TensorHandle* tensor(size_t index) const { return tensors_[index].get(); }
    
      // If the `shape` argument to `FromTensorHandles` is specified, returns that.
      //
      // Otherwise if all of the tensors have the same shape, returns that via the
      // `shape` output argument. This blocks waiting for async tensors, may return
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 25 15:21:13 GMT 2023
    - 12.9K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_experimental.cc

          input_tensors_vector[i] = &input_tensor;
        }
      }
    
      // Create an inference context with dummy values, which will be updated later.
      InferenceContext c(TF_GRAPH_DEF_VERSION, node_def, op_reg_data->op_def,
                         std::vector<ShapeHandle>(num_inputs), input_tensors_vector,
                         {},
                         std::vector<std::unique_ptr<std::vector<ShapeAndType>>>());
    
      // Set input_shapes.
    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)
  3. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      for (int component_index = 0; component_index < tensors_.size();
           ++component_index) {
        // TODO(allenl): Add a C API for summarizing tensors. Currently custom
        // devices limiting themselves to a C API (for ABI compatibility) would need
        // to implement summarization for component tensors themselves.
        ImmediateExecutionTensorHandle* component =
            tensorflow::unwrap(tensors_[component_index].get());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  4. 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)
  5. tensorflow/c/c_api_internal.h

    struct TF_ImportGraphDefOptions {
      tensorflow::ImportGraphDefOptions opts;
    
      // Backing memory for TensorId fields in opts.
      // TODO(skyewm): it'd be better if ImportGraphDefOptions owned this.
      std::vector<tensorflow::string> tensor_id_data;
    };
    
    struct TF_ImportGraphDefResults {
      std::vector<TF_Output> return_tensors;
      std::vector<TF_Operation*> return_nodes;
      std::vector<const char*> missing_unused_key_names;
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sat May 13 00:49:12 GMT 2023
    - 7.6K bytes
    - Viewed (0)
  6. tensorflow/c/eager/immediate_execution_tensor_handle.h

    limitations under the License.
    ==============================================================================*/
    #ifndef TENSORFLOW_C_EAGER_IMMEDIATE_EXECUTION_TENSOR_HANDLE_H_
    #define TENSORFLOW_C_EAGER_IMMEDIATE_EXECUTION_TENSOR_HANDLE_H_
    
    #include "tensorflow/c/eager/abstract_tensor_handle.h"
    #include "tensorflow/c/tensor_interface.h"
    #include "tensorflow/core/framework/types.pb.h"
    #include "tensorflow/core/platform/status.h"
    
    namespace tensorflow {
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Mar 10 21:56:24 GMT 2023
    - 4.3K bytes
    - Viewed (0)
  7. tensorflow/c/c_api_test.cc

      EXPECT_TF_META("v", 2, TF_ATTR_TENSOR, -1);
      TF_Tensor* values[2];
      TF_OperationGetAttrTensorList(oper, "v", &values[0], TF_ARRAYSIZE(values),
                                    s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    
      const char* tensor_data[] = {&tensor1[0], &tensor2[0]};
      const size_t tensor_size[] = {TF_ARRAYSIZE(tensor1), TF_ARRAYSIZE(tensor2)};
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  8. tensorflow/c/eager/c_api_unified_experimental_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
      TF_Tensor* result_tensor = TFE_TensorHandleResolve(result_t, status.get());
    
      // Copy Tensor data into an array.
      float result_data[4] = {0};
      memcpy(&result_data[0], TF_TensorData(result_tensor),
             TF_TensorByteSize(result_tensor));
    
      int data_len = 4;  // length of result_data
      for (int i = 0; i < data_len; i++) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri May 19 21:44:52 GMT 2023
    - 39.1K bytes
    - Viewed (0)
  9. tensorflow/c/experimental/gradients/nn_grad_test.cc

          ctx, inputs[0], inputs[1], &loss, &backprop,
          "SparseSoftmaxCrossEntropyWithLogits"));
      // `gradient_checker` only works with model that returns only 1 tensor.
      // Although, `ops::SparseSoftmaxCrossEntropyWithLogits` returns 2 tensors, the
      // second tensor isn't needed for computing gradient so we could safely drop
      // it.
      outputs[0] = loss;
      backprop->Unref();
      return absl::OkStatus();
    }
    
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 8.3K bytes
    - Viewed (0)
  10. tensorflow/c/eager/unified_api_testutil.h

      TFE_TensorHandle* input_eager =
          TestScalarTensorHandle<T, datatype>(eager_ctx, value);
      *tensor =
          unwrap(TF_CreateAbstractTensorFromEagerTensor(input_eager, status.get()));
      return absl::OkStatus();
    }
    
    // Places data from `t` into *result_tensor.
    Status GetValue(AbstractTensorHandle* t, TF_Tensor** result_tensor);
    }  // namespace tensorflow
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 4K bytes
    - Viewed (0)
Back to top