Lines Matching defs:rhs_shape
989 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
1045 …nsor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, DataTy… in compute_reference()
1143 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
1200 …nsor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, unsign… in compute_reference()
1294 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
1350 …nsor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, DataTy… in compute_reference()
1465 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const Tensor… in compute_target()
1564 …SimpleTensor<T> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, cons… in compute_reference()
1648 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
1711 …nsor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, DataTy… in compute_reference()
1804 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
1861 …nsor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, unsign… in compute_reference()
1931 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
1973 …SimpleTensor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape) in compute_reference()
2029 …TensorType compute_target(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const GEMMLH… in compute_target()
2071 …nsor<int32_t> compute_reference(const TensorShape &lhs_shape, const TensorShape &rhs_shape, unsign… in compute_reference()