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

  /external/tensorflow/tensorflow/compiler/xla/client/lib/
slicing.cc 142 std::vector<XlaOp> to_concat;
144 to_concat.reserve(input_shape.rank());
147 to_concat.push_back(Reshape(index, index_shape.dimensions()));
149 to_concat.push_back(Iota(builder, index_shape, i));
152 XlaOp gather_indices = ConcatInDim(builder, to_concat, input_shape.rank());
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
dynamic_stitch_op.cc 178 std::vector<xla::XlaOp> to_concat(number_of_indices);
186 to_concat[index_num] =
190 ctx->SetOutput(0, xla::ConcatInDim(ctx->builder(), to_concat, 0));
  /external/tensorflow/tensorflow/core/framework/
tensor_util_test.cc 216 std::vector<Tensor> to_concat; local
227 to_concat.push_back(tensor);
233 TF_ASSERT_OK(tensor::Concat(to_concat, &concated));

Completed in 250 milliseconds