Searched refs:ComputeOptions (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.h | 54 struct ComputeOptions;
59 ComputeOptions* compute_opts, bool always_collect_stats);
62 Status SetupArgs(OpKernelContext* ctx, ComputeOptions** compute_opts);
64 Status SetupOutputs(OpKernelContext* ctx, ComputeOptions* opts);
|
| D | map_defun_op.cc | 43 struct MapDefunOp::ComputeOptions { struct in tensorflow::data::MapDefunOp
60 ComputeOptions(OpKernelContext* ctx, OpInputList args, in ComputeOptions() argument
78 MapFunctionCallFrame(ComputeOptions* compute_opts, OpKernel* kernel, in MapFunctionCallFrame()
160 ComputeOptions* const compute_opts_; // Not owned
190 ComputeOptions* compute_opts = nullptr; in ComputeAsync()
203 [](OpKernelContext* ctx, ComputeOptions* compute_opts, DoneCallback& done, in ComputeAsync()
241 ComputeOptions* compute_opts, in SetRunOptions()
256 ComputeOptions** compute_opts) { in SetupArgs()
286 new ComputeOptions(ctx, arguments, captured_inputs, std::move(arg_shapes), in SetupArgs()
291 Status MapDefunOp::SetupOutputs(OpKernelContext* ctx, ComputeOptions* opts) { in SetupOutputs()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | adjust_saturation_op.cc | 45 struct ComputeOptions { struct in tensorflow::AdjustSaturationOpBase
53 const ComputeOptions& options) = 0;
76 ComputeOptions options; in Compute()
184 const ComputeOptions& options) override { in DoCompute()
228 const ComputeOptions& options) override { in DoCompute()
|
| D | adjust_hue_op.cc | 44 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()
|
| D | adjust_contrast_op.cc | 146 struct ComputeOptions { struct in tensorflow::AdjustContrastOpV2Base
176 ComputeOptions options; in Compute()
189 const ComputeOptions& options) = 0;
202 const ComputeOptions& options) override { in DoCompute()
413 const ComputeOptions& options) override { in DoCompute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops.cc | 68 struct ComputeOptions { struct
69 explicit ComputeOptions(OpKernelConstruction* const context) { in ComputeOptions() argument
131 void DoCompute(const ComputeOptions& options, OpKernelContext* const context) { in DoCompute()
249 ComputeOptions options_;
|