Searched refs:model_types (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/ |
| D | keras_parameterized_test.py | 36 model_types = []
46 model_types.append(testing_utils.get_model_type())
54 self.assertLen(model_types, 3)
55 self.assertAllEqual(model_types, [
72 self.assertLen(model_types, 6)
75 model_types = []
89 model_types.append((with_brackets, testing_utils.get_model_type()))
100 self.assertLen(model_types, 6)
101 self.assertAllEqual(model_types, [
121 self.assertLen(model_types, 12)
[all …]
|
| D | keras_parameterized.py | 137 model_types = ['functional', 'subclass', 'sequential']
138 params = [('_%s' % model, model) for model in model_types
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 562 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput() argument
568 if (model_types.HasInputC()) { in ExtractForwardInput()
586 (model_types.rnn_direction_mode == RnnDirectionMode::kRnnBidirectional) in ExtractForwardInput()
617 if (model_types.HasInputC()) { in ExtractForwardInput()
639 const CudnnModelTypes& model_types, bool time_major, in ExtractForwardInput() argument
645 return ExtractForwardInput(context, model_types, time_major, input, input_h, in ExtractForwardInput()
729 const CudnnModelTypes& model_types, in DoForward() argument
757 if (model_types.HasInputC()) { in DoForward()
765 if (model_types.HasInputC()) { in DoForward()
783 model_types.DebugString(), ", ", model_shapes.DebugString()); in DoForward()
[all …]
|