Home
last modified time | relevance | path

Searched refs:core_layers (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/layers/
Dcore_test.py31 from tensorflow.python.layers import core as core_layers unknown
46 dense = core_layers.Dense(2, activation=nn_ops.relu, name='my_dense')
55 dense = core_layers.Dense(2, activation=nn_ops.relu)
58 dense = core_layers.Dense(2, activation=nn_ops.relu)
67 x = core_layers.Dense(1)(v)
73 dense = core_layers.Dense(2, activation=nn_ops.relu, name='my_dense')
92 core_layers.Dense(5)(inputs)
93 core_layers.Dense(2, activation=nn_ops.relu, name='my_dense')(inputs)
97 dense = core_layers.Dense(2, activation=nn_ops.relu, name='my_dense')
104 dense = core_layers.Dense(2, use_bias=False, name='my_dense')
[all …]
Dbase_test.py32 from tensorflow.python.layers import core as core_layers unknown
432 dense = core_layers.Dense(16)
436 dense = core_layers.Dense(16)
645 layer = core_layers.Dense(1)
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Drnn_cells.py25 from tensorflow.contrib.model_pruning.python.layers import core_layers
114 core_layers.MASKED_WEIGHT_NAME)
115 if self._mask not in ops.get_collection_ref(core_layers.MASK_COLLECTION):
116 ops.add_to_collection(core_layers.MASK_COLLECTION, self._mask)
117 ops.add_to_collection(core_layers.MASKED_WEIGHT_COLLECTION,
119 ops.add_to_collection(core_layers.THRESHOLD_COLLECTION, self._threshold)
120 ops.add_to_collection(core_layers.WEIGHT_COLLECTION, self._kernel)
263 core_layers.MASKED_WEIGHT_NAME)
264 if self._mask not in ops.get_collection_ref(core_layers.MASK_COLLECTION):
265 ops.add_to_collection(core_layers.MASK_COLLECTION, self._mask)
[all …]
Dlayers_test.py21 from tensorflow.contrib.model_pruning.python.layers import core_layers
50 masks = ops.get_collection(core_layers.MASK_COLLECTION)
55 masked_weight = ops.get_collection(core_layers.MASKED_WEIGHT_COLLECTION)
75 masks = ops.get_collection(core_layers.MASK_COLLECTION)
83 masked_weight = ops.get_collection(core_layers.MASKED_WEIGHT_COLLECTION)
99 masks = ops.get_collection(core_layers.MASK_COLLECTION)
104 masked_weight = ops.get_collection(core_layers.MASKED_WEIGHT_COLLECTION)
123 masks = ops.get_collection(core_layers.MASK_COLLECTION)
130 masked_weight = ops.get_collection(core_layers.MASKED_WEIGHT_COLLECTION)
Dlayers.py30 from tensorflow.contrib.model_pruning.python.layers import core_layers as core
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Drev_block_lib_test.py27 from tensorflow.python.layers import core as core_layers unknown
48 return core_layers.dense(x, self.CHANNELS // 2, use_bias=True)
51 return core_layers.dense(x, self.CHANNELS // 2, use_bias=True)
71 return core_layers.dense(x, self.CHANNELS // 2, use_bias=True)
74 return core_layers.dense(x, self.CHANNELS // 2, use_bias=True)
102 return core_layers.dense(x, self.CHANNELS // 2, use_bias=True)
107 return core_layers.dense(x, self.CHANNELS // 2, use_bias=True)
169 return core_layers.dense(
177 return core_layers.dense(x, self.CHANNELS // 2)
180 return core_layers.dense(x, self.CHANNELS // 2, activation=nn_ops.relu)
[all …]
Dlayers.py41 from tensorflow.python.layers import core as core_layers unknown
1596 layer = core_layers.Dropout(
1624 outputs = core_layers.flatten(inputs)
1841 layer = core_layers.Dense(
/external/tensorflow/tensorflow/contrib/model_pruning/
DBUILD22 name = "core_layers",
23 srcs = ["python/layers/core_layers.py"],
38 ":core_layers",
70 ":core_layers",
91 ":core_layers",
/external/tensorflow/tensorflow/python/ops/
Dgradients_test.py37 from tensorflow.python.layers import core as core_layers unknown
921 out = core_layers.dense(x, 3, use_bias=False)
946 layer = core_layers.Dense(4, use_bias=False)
974 out = core_layers.dense(x, 4, use_bias=False)
1039 out = core_layers.dense(x, 3, use_bias=False)
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dpruning.py64 from tensorflow.contrib.model_pruning.python.layers import core_layers as core
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_scope_test.py34 from tensorflow.python.layers import core as core_layers unknown
227 d1 = core_layers.Dense(
232 d2 = core_layers.Dense(