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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
dynamic_stitch_op.cc 169 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/
tensor_util_test.cc 150 std::vector<Tensor> to_concat; local
161 to_concat.push_back(tensor);
167 TF_ASSERT_OK(tensor::Concat(to_concat, &concated));

Completed in 117 milliseconds