Searched defs:ComputeOptions (Results 1 – 6 of 6) sorted by relevance
67 struct ComputeOptions { struct68 explicit ComputeOptions(OpKernelConstruction* const context) { in ComputeOptions() argument117 std::unique_ptr<DualLossUpdater> loss_updater;118 int num_sparse_features = 0;119 int num_sparse_features_with_values = 0;120 int num_dense_features = 0;121 int num_inner_iterations = 0;122 int num_loss_partitions = 0;123 bool adaptive = true;124 Regularizations regularizations;
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;
43 struct ComputeOptions { struct in tensorflow::AdjustSaturationOpBase44 const Tensor* input;45 const Tensor* scale;46 Tensor* output;47 int64 channel_count;
44 struct ComputeOptions { struct in tensorflow::AdjustHueOpBase45 const Tensor* input;46 const Tensor* delta;47 Tensor* output;48 int64 channel_count;
132 struct ComputeOptions { struct in tensorflow::data::__anone81855e30111::MapDefunOp137 OpInputList args;138 const std::vector<TensorShape> arg_shapes;139 OpInputList captured_inputs;140 const int64 batch_size;141 std::function<void(std::function<void()>)> runner;146 mutex mu;150 ComputeOptions(OpKernelContext* ctx, OpInputList args, in ComputeOptions() argument
38 struct ComputeOptions { struct in tensorflow::AdjustHsvInYiqOpBase39 const Tensor* input = nullptr;40 Tensor* output = nullptr;41 const Tensor* delta_h = nullptr;42 const Tensor* scale_s = nullptr;43 const Tensor* scale_v = nullptr;44 int64 channel_count = 0;