| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | batch_matmul.h | 76 const int lhs_rows = extended_lhs_shape.Dims(3); in BatchMatMul() local
82 lhs_params.rows = lhs_rows; in BatchMatMul()
92 dst_params.rows = lhs_rows; in BatchMatMul()
106 lhs_rows * rhs_cols; in BatchMatMul()
167 const int lhs_rows = extended_lhs_shape.Dims(3); in BatchMatMul() local
174 const int woff_ext0 = lhs_ext0 == 0 ? 0 : lhs_rows; in BatchMatMul()
175 const int woff_ext1 = lhs_ext1 == 0 ? 0 : lhs_rows; in BatchMatMul()
176 const int woff_ext2 = lhs_ext2 == 0 ? 0 : lhs_rows; in BatchMatMul()
184 lhs_data, row_sums, num_weights_matrices * lhs_rows, accum_depth); in BatchMatMul()
192 lhs_params.rows = lhs_rows; in BatchMatMul()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_matmul.h | 76 const int lhs_rows = extended_lhs_shape.Dims(3); in BatchMatMul() local
91 lhs_rows * rhs_cols; in BatchMatMul()
93 for (int i = 0; i < lhs_rows; ++i) { in BatchMatMul()
99 int idx = lhs_rows * j + i; in BatchMatMul()
134 const int lhs_rows = extended_lhs_shape.Dims(3); in BatchMatMul() local
141 const int woff_ext0 = lhs_ext0 == 0 ? 0 : lhs_rows; in BatchMatMul()
142 const int woff_ext1 = lhs_ext1 == 0 ? 0 : lhs_rows; in BatchMatMul()
143 const int woff_ext2 = lhs_ext2 == 0 ? 0 : lhs_rows; in BatchMatMul()
151 lhs_data, row_sums, num_weights_matrices * lhs_rows, accum_depth); in BatchMatMul()
177 lhs_rows * rhs_cols; in BatchMatMul()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_matmul.cc | 36 tensorflow::int64 lhs_rows = m; in MatMul() local
39 std::swap(lhs_rows, lhs_cols); in MatMul()
48 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> A(lhs, lhs_rows, in MatMul()
|
| D | runtime_matmul.cc | 43 tensorflow::int64 lhs_rows = m; in MatMul() local
46 std::swap(lhs_rows, lhs_cols); in MatMul()
55 const Eigen::TensorMap<Eigen::Tensor<const T, 2>, Alignment> A(lhs, lhs_rows, in MatMul()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 106 auto lhs_rows = lhs.dim_size(ndims_lhs - 2); in Compute() local
111 if (adj_x_) std::swap(lhs_rows, lhs_cols); in Compute()
119 out_shape.AddDim(lhs_rows); in Compute()
|
| /external/llvm-project/mlir/test/Dialect/Linalg/ |
| D | codegen-strategy.mlir | 12 // CHECK-SAME: {lhs_columns = 8 : i32, lhs_rows = 2 : i32, rhs_columns = 4 : i32}
|
| /external/llvm-project/mlir/integration_test/Dialect/Vector/CPU/ |
| D | test-matrix-multiply-row.mlir | 43 { lhs_rows = 2: i32, lhs_columns = 2: i32 , rhs_columns = 2: i32 }
|
| D | test-matrix-multiply-col.mlir | 43 { lhs_rows = 2: i32, lhs_columns = 2: i32 , rhs_columns = 2: i32 }
|
| /external/llvm-project/mlir/include/mlir/Dialect/Vector/ |
| D | VectorOps.td | 2005 I32Attr:$lhs_rows, I32Attr:$lhs_columns, I32Attr:$rhs_columns)>,
2020 The ‘vector.matrix_multiply’ op treats `lhs` as matrix with <lhs_rows> rows
2033 { lhs_rows = 4: i32, lhs_columns = 16: i32 , rhs_columns = 3: i32 } :
2042 $_state.addAttribute("lhs_rows",$_builder.getI32IntegerAttr(lhsRows));
|
| /external/llvm-project/mlir/include/mlir/Dialect/LLVMIR/ |
| D | LLVMOps.td | 998 let arguments = (ins LLVM_Type:$lhs, LLVM_Type:$rhs, I32Attr:$lhs_rows,
1003 $lhs, $rhs, $lhs_rows, $lhs_columns,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 5600 int64 lhs_rows = (spec.lcd == 0) ? (spec.k + k_increase) : spec.m; in TEST_P() local
5602 Shape lhs_shape = ShapeUtil::MakeShape(F32, {lhs_rows, lhs_cols}); in TEST_P()
5605 /*from=*/10.0, /*to=*/10000.0, /*rows=*/lhs_rows, in TEST_P()
5670 int64 lhs_rows = (spec.lcd == 0) ? spec.k : spec.m; in TEST_P() local
5672 Shape lhs_shape = ShapeUtil::MakeShape(F32, {lhs_rows, lhs_cols}); in TEST_P()
5675 /*from=*/10.0, /*to=*/10000.0, /*rows=*/lhs_rows, in TEST_P()
|
| /external/llvm-project/mlir/test/Target/ |
| D | llvmir-intrinsics.mlir | 224 { lhs_rows = 4: i32, lhs_columns = 16: i32 , rhs_columns = 3: i32} :
|
| /external/llvm-project/mlir/lib/Conversion/VectorToLLVM/ |
| D | ConvertVectorToLLVM.cpp | 313 adaptor.rhs(), matmulOp.lhs_rows(), matmulOp.lhs_columns(), in matchAndRewrite()
|
| /external/llvm-project/mlir/test/Conversion/VectorToLLVM/ |
| D | vector-to-llvm.mlir | 834 { lhs_rows = 4: i32, lhs_columns = 16: i32 , rhs_columns = 3: i32 } :
840 // CHECK-SAME: lhs_columns = 16 : i32, lhs_rows = 4 : i32, rhs_columns = 3 : i32
|
| /external/llvm-project/mlir/test/Dialect/Vector/ |
| D | vector-contract-transforms.mlir | 524 …:.*]] = vector.matrix_multiply %[[a3]], %[[b7]] {lhs_columns = 4 : i32, lhs_rows = 2 : i32, rhs_co…
|