Home
last modified time | relevance | path

Searched refs:scale_s (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/image/kernels/
Dadjust_hsv_in_yiq_op.cc42 const Tensor* scale_s = nullptr; member
53 const Tensor& scale_s = context->input(2); in Compute() local
61 OP_REQUIRES(context, TensorShapeUtils::IsScalar(scale_s.shape()), in Compute()
63 scale_s.shape().DebugString())); in Compute()
82 options.scale_s = &scale_s; in Compute()
107 const float scale_s = options.scale_s->scalar<float>()(); in DoCompute() local
112 delta_h, scale_s, scale_v, tranformation_matrix); in DoCompute()
156 const float* scale_s = options.scale_s->flat<float>().data(); in DoCompute() local
159 delta_h, scale_s, scale_v, options.output); in DoCompute()
Dadjust_hsv_in_yiq_op.h37 const float delta_h, const float scale_s, const float scale_v, in compute_tranformation_matrix() argument
55 float vsu = scale_v * scale_s * std::cos(delta_h); in compute_tranformation_matrix()
56 float vsw = scale_v * scale_s * std::sin(delta_h); in compute_tranformation_matrix()
77 const float* const scale_s, const float* const scale_v,
Dadjust_hsv_in_yiq_op_gpu.cu.cc28 const float* const scale_s, in compute_tranformation_matrix_cuda() argument
34 *delta_h, *scale_s, *scale_v, matrix); in compute_tranformation_matrix_cuda()
44 const float* const scale_s, in operator ()() argument
59 1, 0, cu_stream, delta_h, scale_s, scale_v, in operator ()()