OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:Tnum_elements
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
queue_ops.cc
240
const Tensor&
Tnum_elements
= ctx->input(1);
241
int32 num_elements =
Tnum_elements
.flat<int32>()(0);
326
const Tensor&
Tnum_elements
= ctx->input(1);
327
int32 num_elements =
Tnum_elements
.flat<int32>()(0);
barrier_ops.cc
584
const Tensor*
Tnum_elements
;
585
OP_REQUIRES_OK_ASYNC(ctx, ctx->input("num_elements", &
Tnum_elements
),
587
OP_REQUIRES_ASYNC(ctx, TensorShapeUtils::IsScalar(
Tnum_elements
->shape()),
590
const int32 num_elements =
Tnum_elements
->scalar<int32>()();
Completed in 121 milliseconds