OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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