Searched refs:example_index (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | input_target.h | 30 virtual int32 GetTargetAsClassIndex(int example_index,
33 virtual float GetTargetWeight(int example_index) const = 0;
35 virtual float GetTargetAsContinuous(int example_index,
63 int32 GetTargetAsClassIndex(int example_index, in GetTargetAsClassIndex() argument
66 GetTargetAsContinuous(example_index, target_index)); in GetTargetAsClassIndex()
69 float GetTargetWeight(int example_index) const override { in GetTargetWeight() argument
71 return num_weights > 0 && example_index < num_weights in GetTargetWeight()
72 ? (*weight_)(example_index) in GetTargetWeight()
76 float GetTargetAsContinuous(int example_index, in GetTargetAsContinuous() argument
79 return (*target_)(example_index * num_targets_ + target_index); in GetTargetAsContinuous()
|
| D | test_utils.h | 32 int32 GetTargetAsClassIndex(int example_index, in GetTargetAsClassIndex() argument
35 GetTargetAsContinuous(example_index, target_index)); in GetTargetAsClassIndex()
38 float GetTargetWeight(int example_index) const override { in GetTargetWeight() argument
40 return num_weights > 0 && example_index < num_weights in GetTargetWeight()
41 ? (*weight_)[example_index] in GetTargetWeight()
45 float GetTargetAsContinuous(int example_index, in GetTargetAsContinuous() argument
48 return (*target_)[example_index * num_targets_ + target_index]; in GetTargetAsContinuous()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops.cc | 175 const int64 example_index = examples.sampled_index(++atomic_index); in DoCompute() local
176 const Example& example = examples.example(example_index); in DoCompute()
177 const float dual = example_state_data(example_index, 0); in DoCompute()
211 example_state_data(example_index, 0) = new_dual; in DoCompute()
212 example_state_data(example_index, 1) = in DoCompute()
215 example_state_data(example_index, 2) = in DoCompute()
218 example_state_data(example_index, 3) = example_weight; in DoCompute()
|
| D | sdca_internal.h | 319 const Example& example(const int example_index) const { in example() argument
320 return examples_.at(example_index); in example()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/common/stats/ |
| D | gradient-stats.h | 127 GradientStats(const Tensor& g, const Tensor& h, int64 example_index) in GradientStats()
128 : first(g.Slice(example_index, example_index + 1)), in GradientStats()
129 second(h.Slice(example_index, example_index + 1)) {} in GradientStats()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 531 const size_t example_index, const Config& config, in FastParseSerializedExample() argument
585 ", Index: ", example_index, ". ", suffix); in FastParseSerializedExample()
600 if (dense_feature_last_example[d] == example_index) { in FastParseSerializedExample()
604 dense_feature_last_example[d] = example_index; in FastParseSerializedExample()
623 const std::size_t offset = example_index * num_elements; in FastParseSerializedExample()
746 if (sparse_feature_last_example[d] == example_index) { in FastParseSerializedExample()
750 sparse_feature_last_example[d] = example_index; in FastParseSerializedExample()
816 if (dense_feature_last_example[d] == example_index) continue; in FastParseSerializedExample()
826 const std::size_t offset = example_index * num_elements; in FastParseSerializedExample()
852 if (dense_feature_last_example[d] == example_index) continue; in FastParseSerializedExample()
[all …]
|