Searched refs:lhs_transposed (Results 1 – 3 of 3) sorted by relevance
294 Tensor lhs_transposed; in ComputeWithGtsvBatched() local295 TransposeLhsForGtsvBatched(context, lhs, lhs_transposed); in ComputeWithGtsvBatched()297 const Scalar* lhs_data = lhs_transposed.flat<Scalar>().data(); in ComputeWithGtsvBatched()329 Tensor& lhs_transposed) { in TransposeLhsForGtsvBatched() argument351 lhs_transposed_shape, &lhs_transposed)); in TransposeLhsForGtsvBatched()356 &lhs_transposed)); in TransposeLhsForGtsvBatched()
88 HloInstruction* lhs_transposed = builder.AddInstruction( in TEST_P() local91 builder.AddInstruction(CreateCanonicalDot(param_shape, lhs_transposed, rhs)); in TEST_P()
475 auto lhs_transposed = rewriter.create<mhlo::TransposeOp>( in ConvertDot() local511 lhs_transposed.getResult()); in ConvertDot()