| /external/tensorflow/tensorflow/contrib/kfac/python/kernel_tests/ |
| fisher_factors_test.py | 446 has_bias,
452 factor = ff.FullyConnectedKroneckerFactor((tensor,), has_bias=has_bias)
469 factor = ff.FullyConnectedKroneckerFactor((tensor,), has_bias=True)
493 tensor, (1, 2, 3, 4), 3, 2, has_bias=False)
502 tensor, (1, 2, 3, 4), 3, 2, has_bias=True)
512 tensor, (1, 2, 3, 4), 3, 2, has_bias=True)
524 tensor, (1, 2, 1, 1), [1, 1, 1, 1], 'SAME', has_bias=True)
588 factor = ff.FullyConnectedMultiKF((tensor_list,), has_bias=False)
597 factor = ff.FullyConnectedMultiKF((tensor_list,), has_bias=False
[all...] |
| fisher_blocks_test.py | 259 def fisherApprox(self, has_bias=False):
261 if has_bias:
365 lc.LayerCollection(), has_bias=isinstance(params, (tuple, list)))
455 block = fb.FullyConnectedKFACBasicFB(lc.LayerCollection(), has_bias=True)
466 block = fb.FullyConnectedKFACBasicFB(lc.LayerCollection(), has_bias=False)
477 block = fb.FullyConnectedKFACBasicFB(lc.LayerCollection(), has_bias=False)
503 block = fb.FullyConnectedKFACBasicFB(lc.LayerCollection(), has_bias=False)
526 block = fb.FullyConnectedKFACBasicFB(lc.LayerCollection(), has_bias=False)
573 def fisherApprox(self, has_bias=False):
575 if has_bias
[all...] |
| /external/tensorflow/tensorflow/contrib/kfac/python/ops/ |
| fisher_blocks.py | 338 def __init__(self, layer_collection, has_bias=False):
344 has_bias: Whether the component Kronecker factors have an additive bias.
349 self._has_bias = has_bias
668 def __init__(self, layer_collection, has_bias=False):
674 has_bias: Whether the component Kronecker factors have an additive bias.
679 self._has_bias = has_bias
866 def __init__(self, layer_collection, inputs, outputs, has_bias=False):
875 has_bias: bool. If True, estimates Fisher with respect to a bias
884 self._has_bias = has_bias
939 has_bias=False
[all...] |
| layer_collection.py | 506 has_bias = isinstance(params, (tuple, list))
508 block = self.register_block(params, block_type(self, has_bias), reuse=reuse)
615 has_bias = isinstance(params, (tuple, list))
627 block_type(self, inputs, outputs, has_bias=has_bias),
|
| fisher_factors.py | 849 has_bias=False):
857 has_bias: bool. If True, append '1' to each input.
860 self._has_bias = has_bias
916 has_bias=False):
929 has_bias: Python bool. If True, the layer is assumed to have a bias
936 self._has_bias = has_bias
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| export_tensorflow.cc | 331 const bool has_bias = src_op.inputs.size() >= 3; local
333 if (has_bias) {
364 if (has_bias) {
391 const bool has_bias = src_op.inputs.size() >= 3; local
393 if (has_bias) {
444 if (has_bias) {
510 const bool has_bias = src_op.inputs.size() >= 3; local
512 if (has_bias) {
545 if (has_bias) {
[all...] |
| import_tensorflow.cc | 1498 bool has_bias = (input_size == 4); local
[all...] |
| /external/mesa3d/src/gallium/drivers/freedreno/ir3/ |
| ir3_compiler_nir.c | 1394 bool has_bias = false, has_lod = false, has_proj = false, has_off = false; local
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| propagate_fixed_sizes.cc | [all...] |
