Searched refs:count_weights (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/ |
| D | category_encoding.py | 159 def call(self, inputs, count_weights=None): argument
165 if count_weights is not None and self.output_mode != COUNT:
187 return sparse_bincount(inputs, out_depth, binary_output, count_weights)
189 return dense_bincount(inputs, out_depth, binary_output, count_weights)
192 def sparse_bincount(inputs, out_depth, binary_output, count_weights=None): argument
196 weights=count_weights,
210 def dense_bincount(inputs, out_depth, binary_output, count_weights=None): argument
214 weights=count_weights,
|
| D | category_encoding_test.py | 123 int_data = layer(input_data, count_weights=weight_data)
188 int_data = layer(input_data, count_weights=weight_data)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.experimental.preprocessing.-category-encoding.pbtxt | 177 …argspec: "args=[\'self\', \'inputs\', \'count_weights\'], varargs=None, keywords=None, defaults=[\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.layers.experimental.preprocessing.-category-encoding.pbtxt | 177 …argspec: "args=[\'self\', \'inputs\', \'count_weights\'], varargs=None, keywords=None, defaults=[\…
|