Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/image/
Dadjust_saturation_op.cc87 class AdjustSaturationOp;
178 class AdjustSaturationOp<CPUDevice, float> : public AdjustSaturationOpBase { class
180 explicit AdjustSaturationOp(OpKernelConstruction* context) in AdjustSaturationOp() function in tensorflow::AdjustSaturationOp
218 AdjustSaturationOp<CPUDevice, float>);
222 class AdjustSaturationOp<GPUDevice, T> : public AdjustSaturationOpBase { class
224 explicit AdjustSaturationOp(OpKernelConstruction* context) in AdjustSaturationOp() function in tensorflow::AdjustSaturationOp
249 AdjustSaturationOp<GPUDevice, T>);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_ops.cc219 class AdjustSaturationOp : public XlaOpKernel { class
221 explicit AdjustSaturationOp(OpKernelConstruction* context) in AdjustSaturationOp() function in tensorflow::__anonca00298a0111::AdjustSaturationOp
272 REGISTER_XLA_OP(Name("AdjustSaturation"), AdjustSaturationOp);
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_with_tf2xla.cc96 TypeID::get<TF::AdjustSaturationOp>(), in IsOpAllowedTf2XlaFallback()