Searched refs:to_concatenate (Results 1 – 2 of 2) sorted by relevance
31 DatasetBase* to_concatenate, DatasetBase** output) override { in MakeDataset() argument32 OP_REQUIRES(ctx, input->output_dtypes() == to_concatenate->output_dtypes(), in MakeDataset()37 DataTypeVectorString(to_concatenate->output_dtypes())))); in MakeDataset()38 *output = new Dataset(ctx, input, to_concatenate); in MakeDataset()45 const DatasetBase* to_concatenate) in Dataset() argument48 to_concatenate_(to_concatenate) { in Dataset()53 auto os_concatenate = to_concatenate->output_shapes(); in Dataset()
346 std::vector<Tensor> to_concatenate; in ConcatInputTensors() local347 to_concatenate.reserve(batch.num_tasks()); in ConcatInputTensors()349 to_concatenate.push_back(batch.task(task_idx).inputs.at(i)); in ConcatInputTensors()381 to_concatenate.push_back(padding); in ConcatInputTensors()385 const DataType type = to_concatenate[0].dtype(); in ConcatInputTensors()392 Concat<type>(context, to_concatenate, &concatenated_tensor); \ in ConcatInputTensors()