Searched defs:ComputeOptions (Results 1 – 5 of 5) sorted by relevance
68 struct ComputeOptions { struct69 explicit ComputeOptions(OpKernelConstruction* const context) { in ComputeOptions() function118 std::unique_ptr<DualLossUpdater> loss_updater;119 int num_sparse_features = 0;120 int num_sparse_features_with_values = 0;121 int num_dense_features = 0;122 int num_inner_iterations = 0;123 int num_loss_partitions = 0;124 bool adaptive = true;125 Regularizations regularizations;
43 struct MapDefunOp::ComputeOptions { struct in tensorflow::data::MapDefunOp47 OpInputList args;48 const std::vector<TensorShape> arg_shapes;49 OpInputList captured_inputs;50 const int64_t batch_size;51 std::function<void(std::function<void()>)> runner;56 mutex mu;60 ComputeOptions(OpKernelContext* ctx, OpInputList args, in ComputeOptions() argument
155 struct ComputeOptions { struct in tensorflow::AdjustContrastOpV2Base156 const Tensor* input = nullptr;157 const Tensor* factor = nullptr;158 Tensor* output = nullptr;159 int64_t batch = 0;160 int64_t height = 0;161 int64_t width = 0;162 int64_t channels = 0;
44 struct ComputeOptions { struct in tensorflow::AdjustHueOpBase45 const Tensor* input;46 const Tensor* delta;47 Tensor* output;48 int64_t channel_count;
45 struct ComputeOptions { struct in tensorflow::AdjustSaturationOpBase46 const Tensor* input;47 const Tensor* scale;48 Tensor* output;49 int64_t channel_count;