- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for type (0.14 sec)
-
tensorflow/c/eager/c_api_experimental.h
// off, and an execution operation for named operations. Often execution will // simply wrap op execution on one or more physical devices. // // device_info is an opaque caller-defined type stored with the custom device // which is passed to the functions referenced in the TFE_CustomDevice struct // `device` (execute, delete_device, etc.). It can for example contain the // names of wrapped devices. //
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Viewed (0) -
tensorflow/c/eager/tape.h
tensor_stack.pop_back(); auto op_id_it = tensor_tape.find(tensor_id); if (op_id_it == tensor_tape.end()) { continue; } int64_t op_id = op_id_it->second; auto op_it = op_tape->find(op_id); auto result_op_it = result.op_tape.find(op_id); if (op_id == -1 || op_it == op_tape->end() || result_op_it != result.op_tape.end()) { continue; }
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) -
tensorflow/c/eager/c_api_test_util.h
TFE_TensorHandle* axis); // If there is a device of type `device_type`, returns true // and sets 'device_name' accordingly. // `device_type` must be either "GPU" or "TPU". bool GetDeviceName(TFE_Context* ctx, tensorflow::string* device_name, const char* device_type); // Create a ServerDef with the given `job_name` and add `num_tasks` tasks in it.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Jul 17 23:43:59 GMT 2023 - 7.7K bytes - Viewed (0) -
tensorflow/c/c_api.h
// Length of the list if is_list is true. Undefined otherwise. int64_t list_size; // Type of elements of the list if is_list != 0. // Type of the single value stored in the attribute if is_list == 0. TF_AttrType type; // Total size the attribute value. // The units of total_size depend on is_list and type. // (1) If type == TF_ATTR_STRING and is_list == 0 // then total_size is the byte size of the string
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K bytes - Viewed (3) -
tensorflow/c/c_test.c
// A create function. This will never actually get called in this test, it's // just nice to know that it compiles. void* create(TF_OpKernelConstruction* ctx) { TF_DataType type; TF_Status* s = TF_NewStatus(); TF_OpKernelConstruction_GetAttrType(ctx, "foobar", &type, s); TF_DeleteStatus(s); return NULL; } // A compute function. This will never actually get called in this test, it's // just nice to know that it compiles.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Apr 24 20:50:35 GMT 2024 - 2.8K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util.h
Tensor* tensor); absl::StatusOr<xla::PjRtCApiClient*> GetPjRtCApiClient( const DeviceType& device_type); absl::Status ResetPjRtClient(const DeviceType& device_type); } // namespace tensorflow
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 09 19:12:59 GMT 2023 - 1.5K bytes - Viewed (0) -
tensorflow/c/checkpoint_reader.h
std::unique_ptr<tensorflow::Tensor>* out_tensor, TF_Status* out_status) const; private: // Uses "v2_reader_" to build "var name -> shape" and "var name -> data type" // maps; both owned by caller. // REQUIRES: "v2_reader_ != nullptr && v2_reader_.status().ok()". std::pair<std::unique_ptr<TensorSliceReader::VarToShapeMap>,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 12 08:49:52 GMT 2023 - 3.1K bytes - Viewed (0) -
tensorflow/c/eager/unified_api_testutil.h
absl::Span<AbstractTensorHandle*> outputs, bool use_function); Status BuildImmediateExecutionContext(bool use_tfrt, AbstractContext** ctx); // Return a tensor handle with given type, values and dimensions. template <class T, TF_DataType datatype> Status TestTensorHandleWithDims(AbstractContext* ctx, const T* data, const int64_t* dims, int num_dims,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 4K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.h
TF_CAPI_EXPORT extern void TF_CreateAndSetPjRtCApiClient( const char* device_type, TF_Status* status, PJRT_NamedValue* create_options, int num_options); // Resets the PjRt client for a device. After this, `TF_GetPjRtCClient` will // returns an error for that device. TF_CAPI_EXPORT extern void TF_ResetPjRtCClient(const char* device_type, TF_Status* status);
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Dec 20 20:01:06 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_context.h
#define TENSORFLOW_C_EAGER_IMMEDIATE_EXECUTION_CONTEXT_H_ #include <functional> #include <map> #include <memory> #include <vector> #include "absl/types/optional.h" #include "absl/types/span.h" #include "tensorflow/c/eager/abstract_context.h" #include "tensorflow/c/eager/immediate_execution_distributed_manager.h" #include "tensorflow/c/eager/immediate_execution_operation.h"
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 06 08:34:00 GMT 2023 - 12.3K bytes - Viewed (0)