OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:input_num_dims
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/contrib/lite/kernels/
squeeze.cc
44
int
input_num_dims
= NumDimensions(op_context.input);
local
50
TF_LITE_ENSURE(context,
input_num_dims
<= 8);
54
for (int idx = 0; idx <
input_num_dims
; ++idx) {
62
int current = squeeze_dims[idx] < 0 ? squeeze_dims[idx] +
input_num_dims
64
TF_LITE_ENSURE(context, current >= 0 && current <
input_num_dims
&&
72
TfLiteIntArrayCreate(
input_num_dims
- num_squeezed_dims);
73
for (int in_idx = 0, out_idx = 0; in_idx <
input_num_dims
; ++in_idx) {
mean.cc
72
int
input_num_dims
= NumDimensions(op_context->input);
local
75
TfLiteIntArray* output_dims = TfLiteIntArrayCreate(
input_num_dims
);
76
for (int idx = 0; idx <
input_num_dims
; ++idx) {
79
if (axis[axis_idx] == idx || axis[axis_idx] +
input_num_dims
== idx) {
97
current +=
input_num_dims
;
99
TF_LITE_ENSURE(context, current >= 0 && current <
input_num_dims
);
103
previous +=
input_num_dims
;
113
TfLiteIntArrayCreate(
input_num_dims
- num_reduce_axis);
115
for (int idx = 0; idx <
input_num_dims
; ++idx) {
118
if (axis[axis_idx] == idx || axis[axis_idx] +
input_num_dims
== idx)
[
all
...]
Completed in 435 milliseconds