OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:kDims
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
depthtospace_op.cc
76
constexpr int
kDims
= is_int8x4 ? 5 : 4;
77
OP_REQUIRES(context,
kDims
== dims,
78
errors::InvalidArgument("Input rank should be: ",
kDims
,
112
auto Tinput = input.tensor<T,
kDims
>();
113
auto Toutput = outputs_tensor->tensor<T,
kDims
>();
spacetodepth_op.cc
76
constexpr int
kDims
= is_int8x4 ? 5 : 4;
77
OP_REQUIRES(context,
kDims
== dims,
78
errors::InvalidArgument("Input rank should be: ",
kDims
,
114
auto Tinput = input.tensor<T,
kDims
>();
115
auto Toutput = outputs_tensor->tensor<T,
kDims
>();
/external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc
96
const int
kDims
= input_shape.dimensions_count();
98
kDims
> 1 || (
kDims
== 1 && input_shape.dims(0) > 1)) {
99
for (int i = 0; i <
kDims
; ++i) {
[
all
...]
Completed in 874 milliseconds