Home
last modified time | relevance | path

Searched refs:is_int8x4 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddepthtospace_op.cc70 constexpr bool is_int8x4 = std::is_same<T, qint8>::value; in Compute() local
71 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), in Compute()
75 constexpr int kVect = is_int8x4 ? 4 : 1; in Compute()
76 constexpr int kDims = is_int8x4 ? 5 : 4; in Compute()
116 if (is_int8x4) { in Compute()
135 if (!is_int8x4) { in Compute()
Dspacetodepth_op.cc70 constexpr bool is_int8x4 = std::is_same<T, qint8>::value; in Compute() local
71 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), in Compute()
75 constexpr int kVect = is_int8x4 ? 4 : 1; in Compute()
76 constexpr int kDims = is_int8x4 ? 5 : 4; in Compute()
118 if (is_int8x4) { in Compute()
137 if (!is_int8x4) { in Compute()
Dmaxpooling_op.cc1160 constexpr bool is_int8x4 = std::is_same<T, qint8>::value; in Compute() local
1161 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), in Compute()
1172 if (!is_int8x4 && use_dnn_ && data_format_ == FORMAT_NCHW) { in Compute()
1180 if (is_int8x4) { in Compute()
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc125 constexpr bool is_int8x4 = std::is_same<T, qint8>::value; in FusedConv2DBiasActivationOp() local
130 OP_REQUIRES(context, (is_int8x4 == (data_format_ == FORMAT_NCHW_VECT_C)), in FusedConv2DBiasActivationOp()
134 OP_REQUIRES(context, (is_int8x4 == (filter_format_ == FORMAT_OIHW_VECT_I)), in FusedConv2DBiasActivationOp()
420 void AdjustPaddingForCudnn(int padding, bool is_int8x4, int filter_size, in AdjustPaddingForCudnn() argument
424 if (is_int8x4 && filter_size >= 6) { in AdjustPaddingForCudnn()
455 constexpr bool is_int8x4 = std::is_same<T, qint8>::value; in launch() local
456 constexpr int rank = is_int8x4 ? 5 : 4; in launch()
457 constexpr int vect = is_int8x4 ? 4 : 1; in launch()
459 if (is_int8x4) { in launch()
503 AdjustPaddingForCudnn(padding_rows, is_int8x4, filter_rows, &padding_rows, in launch()
[all …]