Searched defs:TensorDimKey (Results 1 – 3 of 3) sorted by relevance
| /dports/misc/py-tvm/incubator-tvm-0.6.1/src/schedule/ |
| H A D | graph.cc | 35 struct TensorDimKey { struct
36 ir::FunctionRef f;
37 int value_index;
38 int dim;
39 TensorDimKey() {} in TensorDimKey() argument
40 TensorDimKey(const ir::Call* op, int dim) in TensorDimKey() function
43 TensorDimKey(const Tensor& t, int dim) in TensorDimKey() function
46 TensorDimKey(const Tensor& t, size_t dim) in TensorDimKey() function
63 struct hash<::tvm::schedule::TensorDimKey> {
|
| /dports/misc/tvm/incubator-tvm-0.6.1/src/schedule/ |
| H A D | graph.cc | 35 struct TensorDimKey { struct
36 ir::FunctionRef f;
37 int value_index;
38 int dim;
39 TensorDimKey() {} in TensorDimKey() function
40 TensorDimKey(const ir::Call* op, int dim) in TensorDimKey() function
43 TensorDimKey(const Tensor& t, int dim) in TensorDimKey() function
46 TensorDimKey(const Tensor& t, size_t dim) in TensorDimKey() function
63 struct hash<::tvm::schedule::TensorDimKey> {
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/te/schedule/ |
| H A D | graph.cc | 38 struct TensorDimKey { struct
42 TensorDimKey() {} in TensorDimKey() function
43 TensorDimKey(const Tensor& t, int dim) : op(t->op), value_index(t->value_index), dim(dim) {} in TensorDimKey() function
44 TensorDimKey(const Tensor& t, size_t dim) in TensorDimKey() function
56 struct hash<::tvm::te::TensorDimKey> {
|