You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Wu Yi 3d750f9c5a [Feature] Fp16 training for resnet50 (#14850) ... * wip * wip * wip * wip for test * add fp16 tests test=develop * fix cpu build test=develop * fix test=develop * fix py3 tests test=develop * fix lr_scheduler dtype test=develop * fix test=dvelop * test fix ci compile test=develop * fix build and merge test=develop * fallback momentumop change to general test=develop		6 years ago
..
details	…
dynload	test=develop	6 years ago
CMakeLists.txt	fix script issue	6 years ago
assert.h	…
cpu_helper.cc	…
cpu_helper.h	…
cpu_helper_test.cc	…
cpu_info.cc	…
cpu_info.h	Remove the useless definition	6 years ago
cpu_info_test.cc	…
cuda_device_function.h	…
cuda_device_guard.cc	…
cuda_device_guard.h	…
cuda_helper_test.cu	…
cuda_primitives.h	…
cuda_profiler.h	…
cudnn_helper.h	…
cudnn_helper_test.cc	…
device_context.cc	Fea/fuse conv elementwise add fuse (#14669)	6 years ago
device_context.h	…
device_context_test.cu	…
device_tracer.cc	…
device_tracer.h	…
enforce.cc	…
enforce.h	…
enforce_test.cc	…
float16.h	…
float16_test.cc	…
float16_test.cu	…
for_range.h	…
gpu_info.cc	…
gpu_info.h	…
hostdevice.h	…
init.cc	…
init.h	…
init_test.cc	…
lock_guard_ptr.h	…
macros.h	…
mkldnn_helper.h	…
mkldnn_reuse.h	…
nccl_helper.h	[Feature] Fp16 training for resnet50 (#14850)	6 years ago
ngraph_helper.h	Enable mul operator for a ngraph engine (#14801)	6 years ago
place.cc	…
place.h	…
place_test.cc	…
port.h	add ctc support for windows	6 years ago
profiler.cc	…
profiler.h	…
profiler.proto	…
profiler_test.cc	…
stream_callback_manager.cc	fix bug	6 years ago
stream_callback_manager.h	…
transform.h	…
transform_test.cu	…
variant.h	…