- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 29 for Type (0.16 sec)
-
tensorflow/c/c_api.cc
metadata.type = TF_ATTR_INT; } else if (typestr == "list(float)") { metadata.type = TF_ATTR_FLOAT; } else if (typestr == "list(bool)") { metadata.type = TF_ATTR_BOOL; } else if (typestr == "list(type)") { metadata.type = TF_ATTR_TYPE; } else if (typestr == "list(shape)") { metadata.type = TF_ATTR_SHAPE;
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/eager/dlpack_test.cc
ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); const DLTensor* dltensor_out = &dlm_out->dl_tensor; EXPECT_EQ(dltensor_out->device.device_type, dltensor_in->device.device_type); EXPECT_EQ(dltensor_out->device.device_id, dltensor_in->device.device_id); EXPECT_EQ(dltensor_out->ndim, dltensor_in->ndim); EXPECT_EQ(dltensor_out->dtype.code, dltensor_in->dtype.code);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
TF_Status* s) { unwrap(o)->outputs.push_back(unwrap(tensor)); } void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type, TF_Status* s) { tsl::Set_TF_Status_from_Status( s, unwrap(op)->Reset(op_type, /*raw_device_name=*/nullptr)); } void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
} void TF_AttrBuilderCheckCanRunOnDevice(TF_AttrBuilder* builder, const char* device_type, TF_Status* status) { status->status = tensorflow::FindKernelDef( tensorflow::DeviceType(device_type), builder->BuildNodeDef(), /* def = */ nullptr, /* kernel_class_name = */ nullptr); }
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/filesystem/plugins/posix/posix_filesystem.cc
if (n > INT32_MAX) requested_read_length = INT32_MAX; else requested_read_length = n; // `pread` returns a `ssize_t` on POSIX, but due to interface being // cross-platform, return type of `Read` is `int64_t`. int64_t r = int64_t{pread(posix_file->fd, dst, requested_read_length, static_cast<off_t>(offset))}; if (r > 0) { dst += r;
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Mar 24 20:08:23 GMT 2024 - 15.8K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
std::string device_type = parsed_name.type; int device_id = 0; if (parsed_name.has_id) { device_id = parsed_name.id; } ctx.device_id = device_id; if (device_type == "CPU") { ctx.device_type = DLDeviceType::kDLCPU; } else if (device_type == "GPU") { #if TENSORFLOW_USE_ROCM ctx.device_type = DLDeviceType::kDLROCM; #else
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/gradients.cc
gtl::ArraySlice<AbstractTensorHandle*> gradient_tensors) const override; // Calls the passed-in backward function. // op_type is the op's name provided in RecordOperation. Status CallBackwardFunction( const string& op_type, GradientFunction* gradient_function, const std::vector<int64_t>& unneeded_gradients, gtl::ArraySlice<AbstractTensorHandle*> output_gradients,
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/c_test_util.cc
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/eager/custom_device_test.cc
executed = false; TFE_Execute(matmul.get(), &retval, &num_retvals, status.get()); EXPECT_FALSE(executed); ASSERT_FALSE(TF_GetCode(status.get()) == TF_OK); // Custom devices can refuse to do type-based dispatch (as hcustom1 is // configured to do) matmul.reset(MatMulOp(context.get(), hcustom1.get(), hcpu.get())); num_retvals = 1; executed = false;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 18.4K bytes - Viewed (0) -
tensorflow/c/checkpoint_reader.cc
var_to_data_type_map_.swap(result.second); } else { reader_.reset(new TensorSliceReader(filename)); if (!reader_->status().ok()) { tsl::Set_TF_Status_from_Status(status, reader_->status()); return; } var_to_shape_map_.reset( new TensorSliceReader::VarToShapeMap(reader_->GetVariableToShapeMap())); var_to_data_type_map_.reset(new TensorSliceReader::VarToDataTypeMap(
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)