Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op.cc71 ComputeOptions* compute_opts = nullptr; in ComputeAsync()
84 [](OpKernelContext* ctx, ComputeOptions* compute_opts, in ComputeAsync()
132 struct ComputeOptions { struct in tensorflow::data::__anone81855e30111::MapDefunOp
150 ComputeOptions(OpKernelContext* ctx, OpInputList args, in ComputeOptions() function
168 MapFunctionCallFrame(ComputeOptions* compute_opts, OpKernel* kernel, in MapFunctionCallFrame()
245 ComputeOptions* const compute_opts_; // Not owned
252 ComputeOptions* compute_opts, bool always_collect_stats) { in SetRunOptions()
266 Status SetupArgs(OpKernelContext* ctx, ComputeOptions** compute_opts) { in SetupArgs()
295 *compute_opts = new ComputeOptions( in SetupArgs()
301 Status SetupOutputs(OpKernelContext* ctx, ComputeOptions* opts) { in SetupOutputs()
/external/tensorflow/tensorflow/contrib/image/kernels/
Dadjust_hsv_in_yiq_op.cc38 struct ComputeOptions { struct in tensorflow::AdjustHsvInYiqOpBase
48 const ComputeOptions& options) = 0;
79 ComputeOptions options; in Compute()
101 const ComputeOptions& options) override { in DoCompute()
150 void DoCompute(OpKernelContext* ctx, const ComputeOptions& options) override { in DoCompute()
/external/tensorflow/tensorflow/core/kernels/
Dadjust_saturation_op.cc43 struct ComputeOptions { struct in tensorflow::AdjustSaturationOpBase
51 const ComputeOptions& options) = 0;
74 ComputeOptions options; in Compute()
182 const ComputeOptions& options) override { in DoCompute()
226 const ComputeOptions& options) override { in DoCompute()
Dadjust_hue_op.cc44 struct ComputeOptions { struct in tensorflow::AdjustHueOpBase
52 const ComputeOptions& options) = 0;
75 ComputeOptions options; in Compute()
205 const ComputeOptions& options) override { in DoCompute()
260 const ComputeOptions& options) override { in DoCompute()
Dadjust_contrast_op.cc146 struct ComputeOptions { struct in tensorflow::AdjustContrastOpV2Base
176 ComputeOptions options; in Compute()
189 const ComputeOptions& options) = 0;
202 const ComputeOptions& options) override { in DoCompute()
412 const ComputeOptions& options) override { in DoCompute()
441 const ComputeOptions& options) override { in DoCompute()
Dsdca_ops.cc67 struct ComputeOptions { struct
68 explicit ComputeOptions(OpKernelConstruction* const context) { in ComputeOptions() argument
130 void DoCompute(const ComputeOptions& options, OpKernelContext* const context) { in DoCompute()
248 ComputeOptions options_;