Searched refs:layer_name (Results 1 – 1 of 1) sorted by relevance
144 def shape_to_config(self, input_shape, filter_shape, output_shape, layer_name): argument145 if layer_name == "AVERAGE_POOL_2D":148 elif layer_name == "CONV_2D" or layer_name == "DEPTHWISE_CONV_2D":152 elif layer_name == "FULLY_CONNECTED":159 elif layer_name == "SOFTMAX":
