HomeSort by relevance Sort by last modified time
    Searched defs:tensor_size (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/testing/
tflite_driver.cc 91 int tensor_size = tensor.bytes / sizeof(T); local
94 for (int i = 0; i < tensor_size; ++i) {
  /external/tensorflow/tensorflow/core/kernels/
stack_ops.cc 176 const Tensor* tensor_size; variable
177 OP_REQUIRES_OK(ctx, ctx->input("max_size", &tensor_size));
179 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(tensor_size->shape()),
182 tensor_size->shape().DebugString()));
184 int32 size_value = tensor_size->scalar<int32>()();
tensor_array_ops.cc 183 const Tensor* tensor_size; variable
184 TF_RETURN_IF_ERROR(ctx->input("size", &tensor_size));
186 if (!TensorShapeUtils::IsScalar(tensor_size->shape())) {
189 tensor_size->shape().DebugString());
191 const int32 size = tensor_size->scalar<int32>()();
    [all...]
  /external/tensorflow/tensorflow/python/lib/core/
ndarray_tensor.cc 216 const size_t tensor_size = TF_TensorByteSize(src); local
217 const char* limit = static_cast<const char*>(tensor_data) + tensor_size;
227 if (expected_tensor_size - tensor_size) {
231 " strings, but the tensor is encoded in ", tensor_size, " bytes");
  /external/tensorflow/tensorflow/core/grappler/costs/
op_level_cost_estimator.cc 828 int64 tensor_size = 1; local
833 tensor_size *= dim.size();
835 return tensor_size;
    [all...]
virtual_scheduler.cc 973 const auto tensor_size = local
    [all...]
  /external/tensorflow/tensorflow/c/
c_api_test.cc 2145 const size_t tensor_size[] = {TF_ARRAYSIZE(tensor1), TF_ARRAYSIZE(tensor2)}; local
    [all...]

Completed in 408 milliseconds