Searched refs:split_tensor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | split_v_op.cc | 58 const Tensor& split_tensor = context->input(1); in ComputeEasyCases() local
71 split_tensor.dims() == 1 && split_tensor.NumElements() == num_split, in ComputeEasyCases()
74 split_tensor.dims(), " -D and ", in ComputeEasyCases()
75 split_tensor.NumElements(), " elements")); in ComputeEasyCases()
77 auto split_sizes_d = split_tensor.vec<Tlen>(); in ComputeEasyCases()
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 519 std::vector<Tensor> split_tensor; in SplitOutputTensors() local
521 output_tensor, task_sizes_plus_optional_padding, &split_tensor); in SplitOutputTensors()
527 DCHECK_EQ(split_tensor.size(), task_sizes_plus_optional_padding.size()); in SplitOutputTensors()
528 if (split_tensor.size() != task_sizes_plus_optional_padding.size()) { in SplitOutputTensors()
531 split_tensor.size(), " splits; expected ", in SplitOutputTensors()
540 tensor_vector[i] = std::move(split_tensor[j]); in SplitOutputTensors()
542 task.context->set_output(i, split_tensor[j]); in SplitOutputTensors()
|