Searched defs:is_train (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/tests/st/dynamic_shape/ |
| D | test_getnext_dynamic_pipeline.py | 23 def _exec_preprocess(network, is_train, dataset, dataset_sink_mode, sink_size=-1, epoch_num=1, data… argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | batch_norm_relu_fusion.cc | 46 auto is_train = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("is_training"); in Process() local
|
| D | batch_norm_add_relu_fusion.cc | 49 auto is_train = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("is_training"); in Process() local
|
| D | post_batch_norm_add_relu_fusion.cc | 49 auto is_train = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("is_training"); in Process() local
|
| D | batch_norm_relu_grad_fusion.cc | 42 auto is_train = AnfAlgo::GetCNodePrimitive(node)->GetAttr("is_training"); in Process() local
|
| D | batch_norm_add_relu_grad_fusion.cc | 187 auto is_train = AnfAlgo::GetCNodePrimitive(batch_norm)->GetAttr("is_training"); in Process() local
|
| /third_party/mindspore/mindspore/train/train_thor/ |
| D | model_thor.py | 127 def _exec_preprocess(self, network, is_train, phase, dataset, dataset_sink_mode, sink_size=-1, argument
|
| /third_party/mindspore/mindspore/lite/tools/converter/ |
| D | anf_transform.cc | 512 auto is_train = config->trainModel; in StoreBuiltinPass() local
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | model_thor.py | 268 …def _exec_preprocess(self, network, is_train, phase, dataset, dataset_sink_mode, iter_first_order=… argument
|
| /third_party/mindspore/mindspore/train/ |
| D | model.py | 334 …def _exec_preprocess(self, is_train, dataset, dataset_sink_mode, sink_size=-1, epoch_num=1, datase… argument
|