| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | split_test.cc | 30 SplitOpModel(const TensorData& input, int num_splits, in SplitOpModel() argument
38 for (int i = 0; i < num_splits; ++i) { in SplitOpModel()
42 CreateSplitOptions(builder_, num_splits).Union()); in SplitOpModel()
68 void Check(int axis, int num_splits, std::initializer_list<int> input_shape, in Check() argument
75 << " and num_splits=" << num_splits; in Check()
78 SplitOpModel m({TensorType_FLOAT32, input_shape}, num_splits); in Check()
82 for (int i = 0; i < num_splits; ++i) { in Check()
88 SplitOpModel const_m({TensorType_FLOAT32, input_shape}, num_splits, axis); in Check()
91 for (int i = 0; i < num_splits; ++i) { in Check()
|
| D | split.cc | 50 int num_splits) { in ResizeOutputTensors() argument
57 TF_LITE_ENSURE_MSG(context, input_size % num_splits == 0, in ResizeOutputTensors()
59 const int slice_size = input_size / num_splits; in ResizeOutputTensors()
76 TF_LITE_ENSURE_EQ(context, NumOutputs(node), op_context.params->num_splits); in Prepare()
89 op_context.params->num_splits); in Prepare()
104 op_context.params->num_splits)); in Eval()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | tree_utils_test.cc | 97 const int32 num_splits = 3; in TEST() local
108 {num_accumulators, num_splits, num_classes}); in TEST()
115 const int32 num_splits = 3; in TEST() local
127 {num_accumulators, num_splits, num_classes}); in TEST()
134 const int32 num_splits = 3; in TEST() local
150 {num_accumulators, num_splits, num_classes}); in TEST()
159 {num_accumulators, num_splits, num_classes}); in TEST()
168 const int32 num_splits = 3; in TEST() local
185 {num_accumulators, num_splits, num_classes}); in TEST()
192 {num_accumulators, num_splits, num_classes}); in TEST()
|
| D | tree_utils.cc | 85 const int32 num_splits = static_cast<int32>(split_counts.shape().dim_size(1)); in GetTwoBestClassification() local
101 bcast[0] = num_splits; in GetTwoBestClassification()
109 GetTwoBest(num_splits, score_fn, best_score, best_index, second_best_score, in GetTwoBestClassification()
159 const int32 num_splits = static_cast<int32>(split_sums.shape().dim_size(1)); in GetTwoBestRegression() local
182 bcast[0] = num_splits; in GetTwoBestRegression()
186 GetTwoBest(num_splits, in GetTwoBestRegression()
|
| D | tree_utils.h | 278 int32 num_splits) { in IsAllInitialized() argument
279 return features(accumulator, num_splits - 1) >= 0; in IsAllInitialized()
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| D | grow_stats.cc | 188 for (int i = 0; i < num_splits(); ++i) { in AddExample()
223 const int to_remove = num_splits() * prune_fraction_; in CheckPrune()
238 for (int i = 0; i < num_splits(); ++i) { in CheckPrune()
259 std::vector<float> split_scores(num_splits()); in CheckPruneHoeffding()
262 for (int i = 0; i < num_splits(); ++i) { in CheckPruneHoeffding()
276 for (int i = num_splits() - 1; i >= 0; i--) { in CheckPruneHoeffding()
313 GetTwoBest(num_splits(), score_fn, &best_score, &best_index, in CheckFinishEarlyHoeffding()
351 GetTwoBest(num_splits(), score_fn, &best_score, &best_index, in CheckFinishEarlyBootstrap()
385 for (int i = 0; i < num_splits(); ++i) { in BestSplit()
456 for (int split_num = 0; split_num < num_splits(); ++split_num) { in PackToProto()
[all …]
|
| D | graph_collection_operator.cc | 44 for (int i = 0; i < slot->num_splits(); ++i) { in BestSplit()
86 for (int i = 0; i < slot->num_splits(); ++i) { in AddExample()
102 int cand_num = slot->num_splits(); in CreateAndInitializeCandidateWithExample()
|
| D | grow_stats_test.cc | 261 ASSERT_EQ(stats.num_splits(), 2); in TEST()
266 ASSERT_EQ(stats.num_splits(), 2); in TEST()
271 ASSERT_EQ(stats.num_splits(), 2); in TEST()
276 ASSERT_EQ(stats.num_splits(), 2); in TEST()
281 ASSERT_EQ(stats.num_splits(), 1); in TEST()
286 ASSERT_EQ(stats.num_splits(), 1); in TEST()
|
| D | grow_stats.h | 76 int num_splits() const { return splits_.size(); } in num_splits() function
317 left_counts_.resize(num_outputs_ * num_splits()); in ClassificationAddSplitStats()
382 left_counts_.resize(num_splits()); in ClassificationAddSplitStats()
477 left_counts_.resize(num_splits(), stats); in ClassificationAddSplitStats()
478 right_counts_.resize(num_splits(), stats); in ClassificationAddSplitStats()
551 left_sums_.resize(num_outputs_ * num_splits()); in AddSplitStats()
552 left_squares_.resize(num_outputs_ * num_splits()); in AddSplitStats()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 294 int num_splits = c->num_outputs() / 3; in __anon4f27251e0c02() local
296 for (int i = 0; i < num_splits; ++i) in __anon4f27251e0c02()
298 for (int i = 0; i < num_splits; ++i) in __anon4f27251e0c02()
300 for (int i = 0; i < num_splits; ++i) in __anon4f27251e0c02()
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| D | builtin_op_data.h | 199 int num_splits; member
|
| D | model.cc | 541 params->num_splits = schema_params->num_splits(); in ParseOpData()
|
| /external/tensorflow/tensorflow/contrib/lite/schema/ |
| D | schema.fbs | 356 num_splits: int;
|
| D | schema_generated.h | 3417 int32_t num_splits;
3419 : num_splits(0) {
3428 int32_t num_splits() const {
3444 void add_num_splits(int32_t num_splits) {
3445 fbb_.AddElement<int32_t>(SplitOptions::VT_NUM_SPLITS, num_splits, 0);
3461 int32_t num_splits = 0) {
3463 builder_.add_num_splits(num_splits);
5363 { auto _e = num_splits(); _o->num_splits = _e; };
5374 auto _num_splits = _o->num_splits;
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| D | operator.cc | 618 op->num_split = options.num_splits(); in ReadOptions()
|