OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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