- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 29 for DataType (0.22 sec)
-
tensorflow/c/c_api.cc
TF_DataType value) { desc->node_builder.Attr(attr_name, static_cast<DataType>(value)); } void TF_SetAttrTypeList(TF_OperationDescription* desc, const char* attr_name, const TF_DataType* values, int num_values) { desc->node_builder.Attr( attr_name, ArraySlice<const DataType>( reinterpret_cast<const DataType*>(values), num_values)); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/c_test.c
#include "tensorflow/c/kernels.h" // 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; }
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/eager/c_api_unified_experimental_graph.cc
public: explicit GraphTensor(TF_Output output, TF_Graph* graph) : TracingTensorHandle(kGraph), output_(output), graph_(graph) {} tensorflow::DataType DataType() const override { return static_cast<tensorflow::DataType>(TF_OperationOutputType(output_)); } tensorflow::Status Shape( tensorflow::PartialTensorShape* shape) const override { DCHECK(shape != nullptr); TF_Status status;
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/abstract_tensor_handle.cc
if (!s.ok()) { shape_string = "<error computing shape>"; } else { shape_string = shape.DebugString(); } return absl::StrCat("TensorHandle(shape=", shape_string, ", dtype=", DataType_Name(DataType()), ", type=", FullType().DebugString(), ")"); } Status AbstractTensorHandle::TensorHandleStatus() const { // Tensor handles in current runtime don't carry error info and this method
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
"TFE_DeleteTensorHandle", tensorflow::profiler::TraceMeLevel::kInfo); if (h) { tensorflow::unwrap(h)->Unref(); } } TF_DataType TFE_TensorHandleDataType(TFE_TensorHandle* h) { return static_cast<TF_DataType>(tensorflow::unwrap(h)->DataType()); } int TFE_TensorHandleNumDims(TFE_TensorHandle* h, TF_Status* status) { if (h == nullptr) {
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.cc
EXPECT_NE(attr_found, attr_values.cend()); EXPECT_EQ(attr_found->second.list().type(0), tensorflow::DataType::DT_BOOL); EXPECT_EQ(attr_found->second.list().type(1), tensorflow::DataType::DT_FLOAT); EXPECT_EQ(attr_found->second.list().type(2), tensorflow::DataType::DT_INT32); TFE_TensorHandle* retvals[1] = {nullptr}; int num_retvals = 1;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/eager/tape.h
} // Returns whether any tensor in a list of tensors is being watched and has // a trainable dtype. bool ShouldRecord(absl::Span<const int64_t> tensor_ids, absl::Span<const tensorflow::DataType> dtypes) const; // Adds this tensor to the list of watched tensors. // // This is a no-op if the tensor is already being watched either from an
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/immediate_execution_context.h
virtual AbstractTensorInterface* CreateBoolScalar(bool value) = 0; // Tensor creation functions virtual AbstractTensorInterface* CreateTensor( DataType dtype, absl::Span<const int64_t> dim_sizes) = 0; typedef void (*MemoryReleaser)(void* data, size_t len, void* arg); // Create a tensor instance from the given data buffer and description.
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) -
tensorflow/c/eager/c_api_experimental.h
// // Custom devices must use TFE_NewCustomDeviceTensorHandle instead. TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandleFromDeviceMemory( TFE_Context* ctx, const char* device_name, TF_DataType, const int64_t* dims, int num_dims, void* data, size_t len, void (*deallocator)(void* data, size_t len, void* arg), void* deallocator_arg, TF_Status* status);
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/unified_api_testutil.cc
for (auto input : inputs) { PartialTensorShape shape; TF_RETURN_IF_ERROR(input->Shape(&shape)); TF_RETURN_IF_ERROR(dyn_cast<tracing::TracingContext>(ctx)->AddParameter( input->DataType(), shape, &handle)); params->emplace_back(handle); } return absl::OkStatus(); } // Runs `model` maybe wrapped in a function. Status RunModel(Model model, AbstractContext* ctx,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K bytes - Viewed (0)