Searched refs:scale_s (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/contrib/image/kernels/ |
D | adjust_hsv_in_yiq_op.cc | 42 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()
|
D | adjust_hsv_in_yiq_op.h | 37 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,
|
D | adjust_hsv_in_yiq_op_gpu.cu.cc | 28 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 ()()
|