Searched refs:channel_axis (Results 1 – 6 of 6) sorted by relevance
42 Eigen::array<int, 1> channel_axis{{1}}; in operator()44 Eigen::IndexList<Eigen::type2index<1> > channel_axis; in operator() local47 V.device(d) = input_data.maximum(channel_axis); in operator()49 range.device(d) = V - input_data.minimum(channel_axis); in operator()
149 channel_axis = 1151 channel_axis = -1152 if input_shape.dims[channel_axis].value is None:155 input_dim = int(input_shape[channel_axis])178 axes={channel_axis: input_dim})747 channel_axis = 1749 channel_axis = -1750 if input_shape.dims[channel_axis].value is None:753 input_dim = int(input_shape[channel_axis])754 self.input_spec = InputSpec(ndim=4, axes={channel_axis: input_dim})[all …]
604 channel_axis = 1606 channel_axis = -1607 if input_shape[channel_axis] is None:610 input_dim = input_shape[channel_axis]
127 channel_axis = 1 if self.data_format == 'channels_first' else -1128 if tensor_shape.dimension_value(input_shape[channel_axis]) is None:131 input_dim = tensor_shape.dimension_value(input_shape[channel_axis])176 ndim=self.rank + 2, axes={channel_axis: input_dim})
62 self.channel_axis = ('channel', self.channels)66 self.batch_axis, self.row_axis, self.column_axis, self.channel_axis,
2012 channel_axis = self._channel_axis()2014 num_channels = input_shape.dims[channel_axis].value2021 channel_axis: num_channels2100 channel_axis = self._channel_axis()2104 if input_shape.dims[channel_axis].value is None: