Home
last modified time | relevance | path

Searched defs:to_concat (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhere_op.cc74 std::vector<xla::XlaOp> to_concat; in Compile() local
Ddynamic_stitch_op.cc178 std::vector<xla::XlaOp> to_concat(number_of_indices); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing.cc204 std::vector<XlaOp> to_concat; in TorchGather() local
283 std::vector<XlaOp> to_concat; in TorchIndexSelect() local
/external/tensorflow/tensorflow/core/framework/
Dtensor_util_test.cc217 std::vector<Tensor> to_concat; in TEST() local