Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for GetTensorSize (0.14 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/jit/pjrt_tensor_buffer_util.cc 

    #include "tensorflow/core/framework/tensor.h"
#include "tensorflow/core/framework/tensor_shape.h"
#include "tensorflow/core/framework/types.h"
#include "tsl/platform/statusor.h"

namespace tensorflow {

static size_t GetTensorSize(const TensorShape& shape, const DataType dtype) {
  return shape.num_elements() * DataTypeSize(dtype);
}

absl::StatusOr<Tensor> MakeTensorFromPjRtBuffer(
    const DataType dtype, const TensorShape& shape,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Sep 14 18:14:47 UTC 2023
    - 3.5K bytes
    - Viewed (0)
Back to top