HomeSort by relevance Sort by last modified time
    Searched refs:split_tensor (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
split_v_op.cc 58 const Tensor& split_tensor = context->input(1); local
71 split_tensor.dims() == 1 && split_tensor.NumElements() == num_split,
72 errors::InvalidArgument("size of the split_tensor must be 1-D and have "
74 split_tensor.dims(), " -D and ",
75 split_tensor.NumElements(), " elements"));
77 auto split_sizes_d = split_tensor.vec<Tlen>();
batch_kernels.cc 447 std::vector<Tensor> split_tensor; local
449 output_tensor, task_sizes_plus_optional_padding, &split_tensor);
455 DCHECK_EQ(split_tensor.size(), task_sizes_plus_optional_padding.size());
456 if (split_tensor.size() != task_sizes_plus_optional_padding.size()) {
459 split_tensor.size(), " splits; expected ",
465 task.context->set_output(i, split_tensor.at(j));
    [all...]

Completed in 113 milliseconds