Searched defs:num_splits (Results 1 – 14 of 14) sorted by relevance
97 const int32 num_splits = 3; in TEST() local115 const int32 num_splits = 3; in TEST() local134 const int32 num_splits = 3; in TEST() local168 const int32 num_splits = 3; in TEST() local
278 int32 num_splits) { in IsAllInitialized()
85 const int32 num_splits = static_cast<int32>(split_counts.shape().dim_size(1)); in GetTwoBestClassification() local159 const int32 num_splits = static_cast<int32>(split_sums.shape().dim_size(1)); in GetTwoBestRegression() local
46 int64 num_splits; in RaggedTensorToSparseShapeFn() local
48 int num_splits; in RaggedGatherShapeFn() local
293 int num_splits = c->num_outputs() / 3; in __anone6e195410c02() local
30 SplitOpModel(const TensorData& input, int num_splits, in SplitOpModel()69 void Check(int axis, int num_splits, std::initializer_list<int> input_shape, in Check()
32 int num_splits, int axis) { in SplitVOpModel()82 int num_splits = size_splits_data.size(); in Check() local
50 const TfLiteTensor* input, int num_splits) { in ResizeOutputTensors()
120 int num_splits = indices_in.dims() - 1 + params_nested_splits_in.size(); in MakeSplits() local220 int64 num_splits = out_splits[i].size(); in WriteSplits() local
43 int64 num_splits = rt_nested_splits.size(); in BuildRaggedTensorToSparseGraph() local
40 int64 num_splits = PARAMS_RAGGED_RANK + indices_shape.dims() - 1; in BuildRaggedGatherGraph() local
292 int num_splits; member296 int num_splits; member
76 int num_splits() const { return splits_.size(); } in num_splits() function