Searched refs:to_concat (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Ddynamic_stitch_op.cc169 std::vector<xla::ComputationDataHandle> to_concat(number_of_indices);
177 to_concat[index_num] =
181 ctx->SetOutput(0, ctx->builder()->ConcatInDim(to_concat, 0));
/external/tensorflow/tensorflow/core/framework/
H A Dtensor_util_test.cc150 std::vector<Tensor> to_concat; local
161 to_concat.push_back(tensor);
167 TF_ASSERT_OK(tensor::Concat(to_concat, &concated));

Completed in 222 milliseconds