- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for type (0.12 sec)
-
tensorflow/c/eager/dlpack.cc
owner->reference.Unref(); delete owner; } // Converts TF_DATAType to DLPack data type. DLDataType GetDlDataType(TF_DataType data_type, TF_Status* status) { DLDataType dtype; dtype.lanes = 1; dtype.bits = TF_DataTypeSize(data_type) * 8; switch (data_type) { case TF_DataType::TF_BOOL: dtype.code = DLDataTypeCode::kDLBool; break; case TF_DataType::TF_HALF: case TF_DataType::TF_FLOAT:
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K 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/gradients.cc
input_ids[i] = ToId(inputs[i]); input_dtypes[i] = inputs[i]->DataType(); } std::vector<TapeTensor> tape_tensors; tape_tensors.reserve(outputs.size()); for (auto t : outputs) { tape_tensors.push_back(TapeTensor(t)); } GradientTape::RecordOperation( op_name, tape_tensors, input_ids, input_dtypes, [gradient_function]() -> GradientFunction* { return gradient_function; },
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
#include "tensorflow/core/framework/full_type.pb.h" #include "tensorflow/core/framework/shape_inference.h" #include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/lib/llvm_rtti/llvm_rtti.h" #include "tensorflow/core/platform/errors.h" #include "tensorflow/core/platform/strcat.h" #include "tensorflow/core/platform/types.h" using tensorflow::dyn_cast;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 15.4K bytes - Viewed (1) -
tensorflow/c/eager/c_api.cc
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/eager/c_api_test_util.cc
for (int i = 0; i < num_devices; ++i) { const string dev_type(TF_DeviceListType(devices, i, status.get())); CHECK_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); const string dev_name(TF_DeviceListName(devices, i, status.get())); CHECK_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); if (dev_type == device_type) { *device_name = dev_name;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
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/c_api_experimental.cc
} TFE_TensorHandle* TFE_NewTensorHandleFromScalar(TF_DataType data_type, void* data, size_t len, TF_Status* status) { auto dtype = static_cast<tensorflow::DataType>(data_type); DCHECK(tensorflow::DataTypeCanUseMemcpy(dtype)); tensorflow::Tensor tensor(dtype, tensorflow::TensorShape({}));
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) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
tensorflow::DeviceType(device_type), std::move(*pjrt_client)); status->status = s; } void TF_ResetPjRtCClient(const char* device_type, TF_Status* status) { status->status = tensorflow::ResetPjRtClient(tensorflow::DeviceType(device_type)); } PJRT_Client* TF_GetPjRtCClient(const char* device_type, TF_Status* status) { absl::StatusOr<xla::PjRtCApiClient*> pjrt_c_api_client =
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/eager/c_api_distributed_test.cc
" attr {" " key: 'dtype'" " value {" " type: DT_FLOAT" " }" " }" " }" " node_def {" " name: 'read1'" " op: 'ReadVariableOp'" " input: 'var'" " device: '/job:localhost/replica:0/task:1/device:CPU:0'" " attr {" " key: 'dtype'" " value {"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0)