Searched refs:lhs_cols (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 107 auto lhs_cols = lhs.dim_size(ndims_lhs - 1); in Compute() local
111 if (adj_x_) std::swap(lhs_rows, lhs_cols); in Compute()
113 OP_REQUIRES(ctx, lhs_cols == rhs_rows, in Compute()
115 "lhs mismatch rhs shape: ", lhs_cols, " vs. ", rhs_rows, in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_matmul.cc | 37 tensorflow::int64 lhs_cols = k; in MatMul() local
39 std::swap(lhs_rows, lhs_cols); in MatMul()
49 lhs_cols); in MatMul()
|
| D | runtime_matmul.cc | 44 tensorflow::int64 lhs_cols = k; in MatMul() local
46 std::swap(lhs_rows, lhs_cols); in MatMul()
56 lhs_cols); in MatMul()
|
| /external/ruy/ruy/ |
| D | trace.h | 779 int lhs_cols = end[Side::kLhs] - start[Side::kLhs]; \
801 kernel_lhs_cols, kernel_rhs_cols, lhs_cols / kernel_lhs_cols, \
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 5601 int64 lhs_cols = (spec.lcd == 0) ? spec.m : (spec.k + k_increase); in TEST_P() local
5602 Shape lhs_shape = ShapeUtil::MakeShape(F32, {lhs_rows, lhs_cols}); in TEST_P()
5606 /*cols=*/lhs_cols))); in TEST_P()
5671 int64 lhs_cols = (spec.lcd == 0) ? spec.m : spec.k; in TEST_P() local
5672 Shape lhs_shape = ShapeUtil::MakeShape(F32, {lhs_rows, lhs_cols}); in TEST_P()
5676 /*cols=*/lhs_cols))); in TEST_P()
|