Home
last modified time | relevance | path

Searched refs:model_type (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/contrib/receptive_field/python/util/examples/
Drf_benchmark.py82 def _construct_model(model_type='resnet_v1_50'): argument
100 if model_type == 'inception_resnet_v2':
102 elif model_type == 'inception_resnet_v2-same':
105 elif model_type == 'inception_v2':
107 elif model_type == 'inception_v2-no-separable-conv':
110 elif model_type == 'inception_v3':
112 elif model_type == 'inception_v4':
114 elif model_type == 'alexnet_v2':
116 elif model_type == 'vgg_a':
118 elif model_type == 'vgg_16':
[all …]
/external/tensorflow/tensorflow/python/keras/
Dtesting_utils.py221 _thread_local_data.model_type = None
237 previous_value = _thread_local_data.model_type
239 _thread_local_data.model_type = value
243 _thread_local_data.model_type = previous_value
280 if _thread_local_data.model_type is None:
285 return _thread_local_data.model_type
352 model_type = get_model_type()
353 if model_type == 'subclass':
355 if model_type == 'subclass_custom_build':
357 if model_type == 'sequential':
[all …]
Dkeras_parameterized.py146 def decorated(self, model_type, *args, **kwargs): argument
148 if model_type == 'functional':
150 elif model_type == 'subclass':
152 elif model_type == 'sequential':
155 raise ValueError('Unknown model type: %s' % (model_type,))
Dcallbacks_test.py1167 model_type = testing_utils.get_model_type()
1186 self._strip_layer_names(summary_file.histograms, model_type),
1198 model_type = testing_utils.get_model_type()
1217 self._strip_layer_names(summary_file.histograms, model_type),
1224 self._strip_layer_names(summary_file.images, model_type),
1233 def _strip_layer_names(self, summaries, model_type): argument
1251 start_from = 2 if 'subclass' in model_type else 1
Dmodels_test.py407 def _clone_and_build_test_helper(self, model, model_type): argument
411 is_subclassed = (model_type == 'subclass')
/external/deqp-deps/SPIRV-Tools/tools/comp/
Dmarkv.cpp143 spvtools::comp::MarkvModelType model_type = in main() local
173 if (model_type != spvtools::comp::kMarkvModelUnknown) in main()
175 model_type = spvtools::comp::kMarkvModelShaderLite; in main()
177 if (model_type != spvtools::comp::kMarkvModelUnknown) in main()
179 model_type = spvtools::comp::kMarkvModelShaderMid; in main()
181 if (model_type != spvtools::comp::kMarkvModelUnknown) in main()
183 model_type = spvtools::comp::kMarkvModelShaderMax; in main()
212 if (model_type == spvtools::comp::kMarkvModelUnknown) in main()
213 model_type = spvtools::comp::kMarkvModelShaderLite; in main()
223 spvtools::comp::CreateMarkvModel(model_type); in main()
/external/swiftshader/third_party/SPIRV-Tools/tools/comp/
Dmarkv.cpp143 spvtools::comp::MarkvModelType model_type = in main() local
173 if (model_type != spvtools::comp::kMarkvModelUnknown) in main()
175 model_type = spvtools::comp::kMarkvModelShaderLite; in main()
177 if (model_type != spvtools::comp::kMarkvModelUnknown) in main()
179 model_type = spvtools::comp::kMarkvModelShaderMid; in main()
181 if (model_type != spvtools::comp::kMarkvModelUnknown) in main()
183 model_type = spvtools::comp::kMarkvModelShaderMax; in main()
212 if (model_type == spvtools::comp::kMarkvModelUnknown) in main()
213 model_type = spvtools::comp::kMarkvModelShaderLite; in main()
223 spvtools::comp::CreateMarkvModel(model_type); in main()
/external/swiftshader/third_party/SPIRV-Tools/test/comp/
Dmarkv_codec_test.cpp91 void TestEncodeDecode(MarkvModelType model_type, in TestEncodeDecode() argument
94 std::unique_ptr<MarkvModel> model = CreateMarkvModel(model_type); in TestEncodeDecode()
137 void TestEncodeDecodeShaderMainBody(MarkvModelType model_type, in TestEncodeDecodeShaderMainBody() argument
221 TestEncodeDecode(model_type, prefix + body + suffix); in TestEncodeDecodeShaderMainBody()
/external/deqp-deps/SPIRV-Tools/test/comp/
Dmarkv_codec_test.cpp91 void TestEncodeDecode(MarkvModelType model_type, in TestEncodeDecode() argument
94 std::unique_ptr<MarkvModel> model = CreateMarkvModel(model_type); in TestEncodeDecode()
137 void TestEncodeDecodeShaderMainBody(MarkvModelType model_type, in TestEncodeDecodeShaderMainBody() argument
221 TestEncodeDecode(model_type, prefix + body + suffix); in TestEncodeDecodeShaderMainBody()
/external/tensorflow/tensorflow/python/keras/layers/
Dcudnn_recurrent_test.py268 model_nest_level=[1, 2], model_type=['seq', 'func']))
273 model_nest_level, model_type): argument
304 model_type)
306 model_nest_level, model_type)
316 def _make_nested_model(self, input_shape, layer, level=1, model_type='func'): argument
334 if model_type == 'func':
336 elif model_type == 'seq':
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/state_space_models/
Dstructural_ensemble_test.py52 self, sample_every, period, num_samples, model_type, model_args, argument
59 model = model_type(
91 model_type=structural_ensemble.StructuralEnsemble,
Dstate_space_model_test.py633 def _time_dependency_test_template(self, model_type): argument
635 model = model_type()
/external/deqp-deps/SPIRV-Tools/source/comp/
Dmarkv_encoder.h80 (model_->model_type() << 16) | model_->model_version(); in GetMarkvBinary()
Dmarkv_decoder.cpp394 const uint32_t model_type = header_.markv_model >> 16; in DecodeModule() local
396 if (model_type != model_->model_type()) in DecodeModule()
Dmarkv_model.h75 uint32_t model_type() const { return model_type_; } in model_type() function
/external/swiftshader/third_party/SPIRV-Tools/source/comp/
Dmarkv_encoder.h80 (model_->model_type() << 16) | model_->model_version(); in GetMarkvBinary()
Dmarkv_decoder.cpp394 const uint32_t model_type = header_.markv_model >> 16; in DecodeModule() local
396 if (model_type != model_->model_type()) in DecodeModule()
Dmarkv_model.h75 uint32_t model_type() const { return model_type_; } in model_type() function