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

  /external/tensorflow/tensorflow/core/kernels/data/
concatenate_dataset_op.cc 31 DatasetBase* to_concatenate, DatasetBase** output) override {
32 OP_REQUIRES(ctx, input->output_dtypes() == to_concatenate->output_dtypes(),
37 DataTypeVectorString(to_concatenate->output_dtypes()))));
38 *output = new Dataset(ctx, input, to_concatenate);
45 const DatasetBase* to_concatenate)
48 to_concatenate_(to_concatenate) {
53 auto os_concatenate = to_concatenate->output_shapes();
  /external/tensorflow/tensorflow/core/kernels/
batch_kernels.cc 351 std::vector<Tensor> to_concatenate; local
352 to_concatenate.reserve(batch->num_tasks());
354 to_concatenate.push_back(batch->task(task_idx).inputs.at(i));
387 to_concatenate.push_back(padding);
391 const DataType type = to_concatenate[0].dtype();
396 concat_status = Concat<type>(last_task_context, to_concatenate, i); \
    [all...]

Completed in 73 milliseconds