Home
last modified time | relevance | path

Searched refs:_batch_flatten_shape_func (Results 1 – 3 of 3) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/relay/op/nn/
H A D_nn.py934 def _batch_flatten_shape_func(data_shape): function
949 return [_batch_flatten_shape_func(inputs[0])]
/dports/misc/tvm/incubator-tvm-0.6.1/python/tvm/relay/op/nn/
H A D_nn.py913 def _batch_flatten_shape_func(data_shape): function
927 return [_batch_flatten_shape_func(inputs[0])]
/dports/misc/py-tvm/incubator-tvm-0.6.1/python/tvm/relay/op/nn/
H A D_nn.py913 def _batch_flatten_shape_func(data_shape): function
927 return [_batch_flatten_shape_func(inputs[0])]