- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 70 for type (1.48 sec)
-
tensorflow/c/checkpoint_reader.cc
string key(v2_reader_->key()); (*var_to_shape_map)[key] = TensorShape(entry.shape()); (*var_to_data_type_map)[key] = DataType(entry.dtype()); } // The returned pointers are owned by the caller. return std::make_pair(std::move(var_to_shape_map), std::move(var_to_data_type_map)); } } // namespace checkpoint
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Aug 25 21:29:12 GMT 2023 - 5.5K bytes - Viewed (0) -
tensorflow/c/c_api_macros.h
#endif // TF_Bool // Macro used to calculate struct size for maintaining ABI stability across // different struct implementations. #ifndef TF_OFFSET_OF_END #define TF_OFFSET_OF_END(TYPE, MEMBER) \ (offsetof(TYPE, MEMBER) + sizeof(((TYPE *)0)->MEMBER)) #endif // TF_OFFSET_OF_END
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sat May 13 04:44:45 GMT 2023 - 1.6K bytes - Viewed (1) -
tensorflow/c/eager/c_api_experimental_test.cc
TFE_TensorHandle* hcpu = TestMatrixTensorHandle(ctx); const char* device_type = TFE_TensorHandleDeviceType(hcpu, status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); ASSERT_TRUE(absl::StrContains(device_type, "CPU")) << device_type; int device_id = TFE_TensorHandleDeviceID(hcpu, status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
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/c_api_test.cc
TF_OperationDescription* init(string type) { // Construct op_name to match the name used by REGISTER_OP in the // ATTR_TEST_REGISTER calls above. string op_name = "CApiAttributesTestOp"; if (type.find("list(") == 0) { op_name += "List"; type = type.replace(0, 5, ""); type = type.replace(type.size() - 1, 1, ""); } op_name += type; return TF_NewOperation(
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) -
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/c_api_test.cc
name_and_attrs.attr().find("dtype")->second.type()); TF_DeleteBuffer(serialized_attr_values); TFE_Op* var_op_2 = TFE_NewOp(ctx, "VarHandleOp", status); string serialized_dtype; ASSERT_TRUE(name_and_attrs.attr().find("dtype")->second.SerializeToString( &serialized_dtype)); TFE_OpSetAttrValueProto( var_op_2, "dtype",
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/c_api_internal.h
}; #endif // !defined(IS_MOBILE_PLATFORM) && !defined(IS_SLIM_BUILD) namespace tensorflow { // Set the shapes and types of the output's handle. // // The lengths of the arrays pointed to by `shapes`, `ranks`, and `types` must // all be equal to `num_shapes_and_types`. If `ranks[i] != -1`, (i.e., if the // rank is known), then it must be equal to the length of `shapes[i]`; if
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) -
tensorflow/c/experimental/filesystem/modular_filesystem.cc
memset(&option, 0, sizeof(option)); option.name = const_cast<char*>(name.c_str()); TF_Filesystem_Option_Value option_value; memset(&option_value, 0, sizeof(option_value)); option_value.type_tag = TF_Filesystem_Option_Type_Buffer; option_value.num_values = values.size(); std::vector<TF_Filesystem_Option_Value_Union> option_values(values.size()); for (size_t i = 0; i < values.size(); i++) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Sep 06 19:12:29 GMT 2023 - 23.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)