Home
last modified time | relevance | path

Searched refs:kChannelSize (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/contrib/image/kernels/
Dadjust_hsv_in_yiq_op.cc69 context, channels == kChannelSize, in Compute()
105 auto input_data = input->shaped<float, 2>({channel_count, kChannelSize}); in DoCompute()
109 auto output_data = output->shaped<float, 2>({channel_count, kChannelSize}); in DoCompute()
110 float tranformation_matrix[kChannelSize * kChannelSize] = {0}; in DoCompute()
111 internal::compute_tranformation_matrix<kChannelSize * kChannelSize>( in DoCompute()
121 const float* p = input_data.data() + start_channel * kChannelSize; in DoCompute()
122 float* q = output_data.data() + start_channel * kChannelSize; in DoCompute()
124 for (int q_index = 0; q_index < kChannelSize; q_index++) { in DoCompute()
126 for (int p_index = 0; p_index < kChannelSize; p_index++) { in DoCompute()
129 tranformation_matrix[q_index + kChannelSize * p_index]; in DoCompute()
[all …]
Dadjust_hsv_in_yiq_op_gpu.cu.cc32 if (matrix_size == kChannelSize * kChannelSize) { in compute_tranformation_matrix_cuda()
33 compute_tranformation_matrix<kChannelSize * kChannelSize>( in compute_tranformation_matrix_cuda()
48 const uint64 k = kChannelSize; in operator ()()
49 const uint64 n = kChannelSize; in operator ()()
54 DT_FLOAT, TensorShape({kChannelSize * kChannelSize}), in operator ()()
Dadjust_hsv_in_yiq_op.h31 static constexpr int kChannelSize = 3; variable
39 static_assert(MATRIX_SIZE == kChannelSize * kChannelSize, in compute_tranformation_matrix()
/external/tensorflow/tensorflow/core/kernels/
Dadjust_saturation_op.cc187 static const int kChannelSize = 3; in DoCompute() local
188 auto input_data = input->shaped<float, 2>({channel_count, kChannelSize}); in DoCompute()
190 auto output_data = output->shaped<float, 2>({channel_count, kChannelSize}); in DoCompute()
198 const float* p = input_data.data() + start_channel * kChannelSize; in DoCompute()
199 float* q = output_data.data() + start_channel * kChannelSize; in DoCompute()
207 p += kChannelSize; in DoCompute()
208 q += kChannelSize; in DoCompute()
Dadjust_hue_op.cc210 static const int kChannelSize = 3; in DoCompute() local
211 auto input_data = input->shaped<float, 2>({channel_count, kChannelSize}); in DoCompute()
213 auto output_data = output->shaped<float, 2>({channel_count, kChannelSize}); in DoCompute()
221 const float* p = input_data.data() + start_channel * kChannelSize; in DoCompute()
222 float* q = output_data.data() + start_channel * kChannelSize; in DoCompute()
241 p += kChannelSize; in DoCompute()
242 q += kChannelSize; in DoCompute()
/external/grpc-grpc/test/core/tsi/alts/frame_protector/
Dalts_frame_protector_test.cc30 const size_t kChannelSize = 32768; variable
77 static_cast<unsigned char*>(gpr_malloc(kChannelSize)); in alts_test_do_round_trip_check_frames()
97 static_cast<unsigned char*>(gpr_malloc(kChannelSize)); in alts_test_do_round_trip_check_frames()