- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for DefaultPaddedShapeFn (0.17 sec)
-
tensorflow/compiler/jit/xla_device.cc
namespace tensorflow { // Default PaddedShapeFn implementation that simply returns the unpadded // on-device shape. This is accurate for CPU and GPU devices that neither // transpose nor pad tensors. Status DefaultPaddedShapeFn(const Tensor& tensor, xla::Shape* shape) { const tensorflow::XlaTensor* xla_tensor = tensorflow::XlaTensor::FromTensor(&tensor); if (xla_tensor == nullptr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.h
XlaDeviceOpRegistrations* RegisterXlaDeviceKernels(const char* device, const char* jit_device); Status DefaultPaddedShapeFn(const Tensor& tensor, xla::Shape* shape); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K bytes - Viewed (0)