Searched refs:lhs_rows (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dot_decomposer.cc | 67 const int64 lhs_rows = lhs_shape.dimensions(num_batch_dims + 0); in DecomposeBatchDot() local
70 ShapeUtil::MakeShape(lhs_type, {batch_size, lhs_rows, lhs_cols}); in DecomposeBatchDot()
72 ShapeUtil::MakeShape(lhs_type, {1, lhs_rows, lhs_cols}); in DecomposeBatchDot()
74 ShapeUtil::MakeShape(lhs_type, {lhs_rows, lhs_cols}); in DecomposeBatchDot()
108 {i + 1, lhs_rows, lhs_cols}, {1, 1, 1})); in DecomposeBatchDot()
122 ShapeUtil::MakeShape(lhs_type, {lhs_cols, lhs_rows}); in DecomposeBatchDot()
|
| D | algebraic_simplifier_test.cc | 4640 int64 lhs_rows = (spec.lcd == 0) ? (spec.k + k_increase) : spec.m; in TEST_P() local
4642 Shape lhs_shape = ShapeUtil::MakeShape(F32, {lhs_rows, lhs_cols}); in TEST_P()
4645 /*from=*/10.0, /*to=*/10000.0, /*rows=*/lhs_rows, in TEST_P()
4710 int64 lhs_rows = (spec.lcd == 0) ? spec.k : spec.m; in TEST_P() local
4712 Shape lhs_shape = ShapeUtil::MakeShape(F32, {lhs_rows, lhs_cols}); in TEST_P()
4715 /*from=*/10.0, /*to=*/10000.0, /*rows=*/lhs_rows, in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_matmul.cc | 39 int64 lhs_rows = m; in MatMul() local
42 std::swap(lhs_rows, lhs_cols); in MatMul()
51 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> A(lhs, lhs_rows, in MatMul()
|
| D | runtime_matmul.cc | 45 int64 lhs_rows = m; in MatMul() local
48 std::swap(lhs_rows, lhs_cols); in MatMul()
57 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> A(lhs, lhs_rows, in MatMul()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_batch_matmul_op.cc | 78 auto lhs_rows = lhs.dim_size(ndims - 2); in Compute() local
82 if (adj_x_) std::swap(lhs_rows, lhs_cols); in Compute()
89 out_shape.AddDim(lhs_rows); in Compute()
|