Searched defs:concatenate_dimension (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
H A Dreference_util.h223 // Concatenates the lhs and rhs arrays along the concatenate_dimension.
224 // E.g. if concatenate_dimension is 0, the "n1"/height dimension is
229 int concatenate_dimension) {
230 CHECK(0 <= concatenate_dimension && concatenate_dimension < 2);
232 concatenate_dimension == 0 ? lhs.n1() + rhs.n1() : lhs.n1(),
233 concatenate_dimension == 1 ? lhs.n2() + rhs.n2() : lhs.n2());
247 // Concatenates the lhs and rhs 3D arrays along the concatenate_dimension. lhs
252 int concatenate_dimension) {
253 CHECK(0 <= concatenate_dimension
227 Concat2D(const Array2D<T>& lhs, const Array2D<T>& rhs, int concatenate_dimension) argument
250 Concat3D(const Array3D<T>& lhs, const Array3D<T>& rhs, int concatenate_dimension) argument
284 Concat4D(const Array4D<T>& lhs, const Array4D<T>& rhs, int concatenate_dimension) argument
[all...]
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instruction.cc1599 int64 HloInstruction::concatenate_dimension() const { function in class:xla::HloInstruction

Completed in 202 milliseconds