Searched refs:transpose_rhs_ (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgemm_thunk.h74 const bool transpose_rhs_; member in class:xla::gpu::GemmThunk
H A Dgemm_thunk.cc224 transpose_rhs_(transpose_rhs) {}
278 make_descriptor(rhs_data, rhs_shape_, transpose_rhs_);
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Ddot_op_emitter.h140 const bool transpose_rhs_; member in class:xla::cpu::DotOpEmitter
H A Ddot_op_emitter.cc534 transpose_rhs_(transpose_rhs),
583 transpose_rhs_ ^ !mat_mult_dims.rhs_column_major;
749 ShapeUtil::GetDimensionNumber(rhs_shape, transpose_rhs_ ? -1 : -2);
976 bool transpose_rhs = transpose_rhs_;
1007 rhs_shape.dimensions(transpose_rhs_ ? 0 : 1),

Completed in 203 milliseconds