| /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/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()
69 void Check(int axis, int num_splits, std::initializer_list<int> input_shape, in Check() argument
77 << " and num_splits=" << num_splits; in Check()
80 SplitOpModel m({type, input_shape}, num_splits); in Check()
84 for (int i = 0; i < num_splits; ++i) { in Check()
91 SplitOpModel const_m({type, input_shape}, num_splits, axis); in Check()
94 for (int i = 0; i < num_splits; ++i) { in Check()
|
| D | split_v_test.cc | 32 int num_splits, int axis) { in SplitVOpModel() argument
40 for (int i = 0; i < num_splits; ++i) { in SplitVOpModel()
44 CreateSplitVOptions(builder_, num_splits).Union()); in SplitVOpModel()
82 int num_splits = size_splits_data.size(); in Check() local
84 num_splits, kAxisIsATensor); in Check()
89 for (int i = 0; i < num_splits; ++i) { in Check()
95 {TensorType_INT32, size_splits_shape}, num_splits, in Check()
100 for (int i = 0; i < num_splits; ++i) { in Check()
|
| D | split.cc | 50 const TfLiteTensor* input, 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()
91 op_context.params->num_splits); in Prepare()
106 op_context.params->num_splits)); in Eval()
|
| D | split_v.cc | 124 TF_LITE_ENSURE_EQ(context, NumOutputs(node), op_context.params->num_splits); in Prepare()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ragged_conversion_ops.cc | 46 int64 num_splits; in RaggedTensorToSparseShapeFn() local
47 TF_RETURN_IF_ERROR(c->GetAttr<int64>("RAGGED_RANK", &num_splits)); in RaggedTensorToSparseShapeFn()
49 if (num_splits < 1) { in RaggedTensorToSparseShapeFn()
52 ShapeHandle rt_dense_values = c->input(num_splits); in RaggedTensorToSparseShapeFn()
56 for (int64 i = 0; i < num_splits; ++i) { in RaggedTensorToSparseShapeFn()
63 ? c->MakeDim(c->Rank(rt_dense_values) + num_splits) in RaggedTensorToSparseShapeFn()
|
| D | ragged_array_ops.cc | 48 int num_splits; in RaggedGatherShapeFn() local
52 TF_RETURN_IF_ERROR(c->GetAttr<int>("OUTPUT_RAGGED_RANK", &num_splits)); in RaggedGatherShapeFn()
57 c->WithRank(indices, num_splits - PARAMS_RAGGED_RANK + 1, &indices)); in RaggedGatherShapeFn()
71 for (int i = 0; i < num_splits; ++i) { in RaggedGatherShapeFn()
80 c->set_output(num_splits, values); in RaggedGatherShapeFn()
|
| D | sparse_ops.cc | 293 int num_splits = c->num_outputs() / 3; in __anona7ffb82a0c02() local
295 for (int i = 0; i < num_splits; ++i) in __anona7ffb82a0c02()
297 for (int i = 0; i < num_splits; ++i) in __anona7ffb82a0c02()
299 for (int i = 0; i < num_splits; ++i) in __anona7ffb82a0c02()
|
| /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/kernels/ |
| D | ragged_tensor_to_sparse_kernel_test.cc | 43 int64 num_splits = rt_nested_splits.size(); in BuildRaggedTensorToSparseGraph() local
45 .Input(FakeInput(num_splits)) // rt_nested_splits in BuildRaggedTensorToSparseGraph()
47 .Attr("RAGGED_RANK", num_splits) in BuildRaggedTensorToSparseGraph()
|
| D | ragged_gather_op.cc | 120 int num_splits = indices_in.dims() - 1 + params_nested_splits_in.size(); in MakeSplits() local
121 out_splits->assign(num_splits, {0}); in MakeSplits()
220 int64 num_splits = out_splits[i].size(); in WriteSplits() local
222 splits_out.allocate(i, TensorShape({num_splits}), &splits)); in WriteSplits()
|
| D | ragged_gather_op_test.cc | 40 int64 num_splits = PARAMS_RAGGED_RANK + indices_shape.dims() - 1; in BuildRaggedGatherGraph() local
47 .Attr("OUTPUT_RAGGED_RANK", num_splits) in BuildRaggedGatherGraph()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 292 int num_splits; member
296 int num_splits; member
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view_test.cc | 69 for (int num_splits : {1, 2}) { in TEST_F()
72 ops::SparseSplit b(s.WithOpName("b"), a, a, a, a, num_splits); in TEST_F()
89 for (int port_id = 0; port_id <= num_splits * 3; ++port_id) { in TEST_F()
91 if (port_id < num_splits * 3) { in TEST_F()
92 arg_id = port_id / num_splits; in TEST_F()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 513 params->num_splits = schema_params->num_splits(); in ParseOpData()
521 params->num_splits = schema_params->num_splits(); in ParseOpData()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_ops.py | 506 num_splits = self.num_packs
518 split_size = total_grad_size // num_splits
519 split_size_last = total_grad_size - split_size * (num_splits - 1)
520 split_sizes = [split_size] * (num_splits - 1) + [split_size_last]
527 device_grad_packs.append(zip(grad_packs, [None] * num_splits))
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema.fbs | 575 num_splits: int;
579 num_splits: int;
|
| D | schema_generated.h | 5260 int32_t num_splits;
5262 : num_splits(0) {
5271 int32_t num_splits() const {
5287 void add_num_splits(int32_t num_splits) {
5288 fbb_.AddElement<int32_t>(SplitOptions::VT_NUM_SPLITS, num_splits, 0);
5304 int32_t num_splits = 0) {
5306 builder_.add_num_splits(num_splits);
5314 int32_t num_splits;
5316 : num_splits(0) {
5325 int32_t num_splits() const {
[all …]
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | ops_compatibility.md | 929 num_splits: Specifies number of outputs
945 num_splits: Specifies number of outputs
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 1437 op->num_split = options.num_splits(); in ReadOptions()
1467 op->num_split = options.num_splits(); in ReadOptions()
|