Searched refs:split_tensors (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_split_op_test.py | 383 def split_tensors(rt): function
386 splited_rts = split_tensors(rt)
402 def split_tensors(rt, split_lengths): function
409 splited_rts = split_tensors(rt, split_lengths)
470 def split_tensors(rt, split_lengths): function
476 self.evaluate(split_tensors(rt=rt, split_lengths=lengths))
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | cross_device_utils.py | 483 split_tensors = []
486 split_tensors.append(gather_padded_out_tensor[start_pos:start_pos +
488 out_tensor_t = array_ops.concat(split_tensors, 0)
560 split_tensors = []
563 split_tensors.append(all_padded_tensors[start_pos:start_pos +
565 return array_ops.concat(split_tensors, 0)
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/ |
| H A D | batch_resource_base.cc | 556 std::vector<Tensor> split_tensors; in SplitInputTask() local
562 output_task_sizes, &split_tensors); in SplitInputTask()
568 if (split_tensors.size() != output_task_sizes.size()) { in SplitInputTask()
572 split_tensors.size(), " splits; expected ", output_task_sizes.size()); in SplitInputTask()
576 auto moved_tensor_iter = std::next(split_tensors.begin(), j); in SplitInputTask()
604 std::map<string, std::vector<Tensor>> split_tensors; in SplitOutputTensors() local
|