Searched refs:segment_ids_tensor (Results 1 – 3 of 3) sorted by relevance
171 segment_ids_tensor = ragged_factory_ops.constant(181 data_tensor, segment_ids_tensor, num_partitions)189 data_flow_ops.dynamic_partition(data_tensor, segment_ids_tensor,
72 auto* segment_ids_tensor = FindTensorByName( in Preprocess() local113 PopulateTensor(std::vector<int>(kMaxSeqLen, 0), segment_ids_tensor); in Preprocess()
176 TfLiteTensor* segment_ids_tensor = in Preprocess() local281 PopulateTensor(segment_ids, segment_ids_tensor); in Preprocess()