Searched refs:to_concatenate (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op.cc | 39 const DatasetBase* to_concatenate) in Dataset() argument
42 to_concatenate_(to_concatenate) { in Dataset()
47 auto os_concatenate = to_concatenate->output_shapes(); in Dataset()
221 DatasetBase* to_concatenate, in MakeDataset() argument
223 OP_REQUIRES(ctx, input->output_dtypes() == to_concatenate->output_dtypes(), in MakeDataset()
228 DataTypeVectorString(to_concatenate->output_dtypes())))); in MakeDataset()
229 *output = new Dataset(ctx, input, to_concatenate); in MakeDataset()
|
| D | concatenate_dataset_op.h | 35 DatasetBase* to_concatenate, DatasetBase** output) override;
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 343 std::vector<Tensor> to_concatenate; in ConcatInputTensors() local
344 to_concatenate.reserve(batch.num_tasks()); in ConcatInputTensors()
346 to_concatenate.push_back(batch.task(task_idx).inputs.at(i)); in ConcatInputTensors()
366 to_concatenate.push_back(padding); in ConcatInputTensors()
372 Concat(context, to_concatenate, &concatenated_tensor); in ConcatInputTensors()
402 std::vector<Tensor> to_concatenate; in SplitInputTask() local
403 to_concatenate.reserve(output->size()); in SplitInputTask()
405 to_concatenate.push_back(std::move((*output)[j][i])); in SplitInputTask()
408 Concat(op_kernel_context, to_concatenate, &output_tensor); in SplitInputTask()
|