Searched defs:ConcatInDim (Results 1 – 2 of 2) sorted by relevance
540 LocalOp ComputationBuilder::ConcatInDim(absl::Span<const LocalOp> operands, in ConcatInDim() function in xla::swig::ComputationBuilder
852 XlaOp XlaBuilder::ConcatInDim(absl::Span<const XlaOp> operands, in ConcatInDim() function in xla::XlaBuilder2900 XlaOp ConcatInDim(XlaBuilder* builder, absl::Span<const XlaOp> operands, in ConcatInDim() function