Searched refs:set_outputs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | parsing_ops_test.cc | 88 auto set_outputs = [&op](int num_sparse, int num_dense, in TEST() local
110 set_outputs(0 /* num_sparse */, 0 /* num_dense */); in TEST()
117 set_outputs(2 /* num_sparse */, 3 /* num_dense */); in TEST()
126 set_outputs(2, 3, true /* add_extra_shape */); in TEST()
131 set_outputs(2, 3, false /* add_extra_shape */, 1 /* unknown_outer_dims */); in TEST()
139 set_outputs(2, 3, true /* add_extra_shape */, 1 /* unknown_outer_dims */); in TEST()
144 set_outputs(2, 3, false /* add_extra_shape */, 2 /* unknown_outer_dims */); in TEST()
151 auto set_outputs = [&op](int num_context_sparse, int num_context_dense, in TEST() local
194 set_outputs(0, 0, 0, 0); in TEST()
201 set_outputs(2 /* num_context_sparse */, 3 /* num_context_dense */, 0, 0); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 432 def set_outputs(self, outputs): member in AutoAddUpdates
|
| D | base_layer.py | 607 auto_updater.set_outputs(outputs)
|