| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| fused_conv2d_bias_activation_op.cc | 118 constexpr bool is_int8x4 = std::is_same<T, qint8>::value; local
123 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)),
127 OP_REQUIRES(context, (is_int8x4 == (filter_format_ == FORMAT_OIHW_VECT_I)),
285 void AdjustPaddingForCudnn(int padding, bool is_int8x4, int filter_size,
289 if (is_int8x4 && filter_size >= 6) {
320 constexpr bool is_int8x4 = std::is_same<T, qint8>::value; local
321 constexpr int rank = is_int8x4 ? 5 : 4;
322 constexpr int vect = is_int8x4 ? 4 : 1;
324 if (is_int8x4) {
368 AdjustPaddingForCudnn(padding_rows, is_int8x4, filter_rows, &padding_rows
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| depthtospace_op.cc | 70 constexpr bool is_int8x4 = std::is_same<T, qint8>::value; variable
71 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)),
75 constexpr int kVect = is_int8x4 ? 4 : 1;
76 constexpr int kDims = is_int8x4 ? 5 : 4;
116 if (is_int8x4) {
135 if (!is_int8x4) {
|
| spacetodepth_op.cc | 70 constexpr bool is_int8x4 = std::is_same<T, qint8>::value; variable
71 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)),
75 constexpr int kVect = is_int8x4 ? 4 : 1;
76 constexpr int kDims = is_int8x4 ? 5 : 4;
118 if (is_int8x4) {
137 if (!is_int8x4) {
|
| maxpooling_op.cc | 1073 constexpr bool is_int8x4 = std::is_same<T, qint8>::value; variable
[all...] |
