Home
last modified time | relevance | path

Searched refs:get_channel_dim (Results 1 – 2 of 2) sorted by relevance

/dports/misc/mmdnn/MMdnn-0.3.1/mmdnn/conversion/examples/tensorflow/models/
H A Dnasnet_utils.py63 def get_channel_dim(shape, data_format=INVALID): function
267 prev_num_filters = get_channel_dim(prev_layer.shape)
349 input_filters = get_channel_dim(net.shape)
376 final_num_filters = get_channel_dim(net[-1].shape)
380 curr_num_filters = get_channel_dim(net[idx].shape)
H A Dnasnet.py314 nasnet_utils.get_channel_dim],
361 nasnet_utils.get_channel_dim],
409 nasnet_utils.get_channel_dim],