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 batch_size;51 std::function<void(std::function<void()>)> runner;56 mutex mu;60 ComputeOptions(OpKernelContext* ctx, OpInputList args, in ComputeOptions() function
146 struct ComputeOptions { struct in tensorflow::AdjustContrastOpV2Base147 const Tensor* input = nullptr;148 const Tensor* factor = nullptr;149 Tensor* output = nullptr;150 int64 batch = 0;151 int64 height = 0;152 int64 width = 0;153 int64 channels = 0;
45 struct ComputeOptions { struct in tensorflow::AdjustSaturationOpBase46 const Tensor* input;47 const Tensor* scale;48 Tensor* output;49 int64 channel_count;
44 struct ComputeOptions { struct in tensorflow::AdjustHueOpBase45 const Tensor* input;46 const Tensor* delta;47 Tensor* output;48 int64 channel_count;