Home
last modified time | relevance | path

Searched refs:execution_mode (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
Dresnet50_test.py95 def _apply(self, defun=False, execution_mode=None): argument
100 with tf.device(device), context.execution_mode(execution_mode):
110 self._apply(defun=False, execution_mode=context.ASYNC)
116 self._apply(defun=True, execution_mode=context.ASYNC)
192 def _test_train(self, execution_mode=None): argument
201 with tf.device(device), context.execution_mode(execution_mode):
216 self._test_train(execution_mode=context.ASYNC)
291 execution_mode=None): argument
292 with context.execution_mode(execution_mode):
304 if execution_mode:
[all …]
/external/tensorflow/tensorflow/python/eager/
Dremote_benchmarks_test.py45 def run_benchmark(func, num_iters, execution_mode=None): argument
47 with context.execution_mode(execution_mode):
50 if execution_mode == context.ASYNC:
55 if execution_mode == context.ASYNC:
87 def _run(self, func, num_iters=1000, execution_mode=context.ASYNC): argument
88 total_time = run_benchmark(func, num_iters, execution_mode)
193 self._run(func, execution_mode=context.ASYNC, num_iters=100)
Dbenchmarks_test.py146 def run_benchmark(func, num_iters, execution_mode=None): argument
148 with context.execution_mode(execution_mode):
151 if execution_mode == context.ASYNC:
156 if execution_mode == context.ASYNC:
175 def _run(self, func, num_iters, execution_mode=None): argument
176 total_time = run_benchmark(func, num_iters, execution_mode)
419 execution_mode=None): argument
421 self._run(func, num_iters, execution_mode=execution_mode)
456 execution_mode=None): argument
459 self._run(func, num_iters, execution_mode=execution_mode)
[all …]
Dcontext.py346 execution_mode=None, argument
404 if execution_mode not in (None, SYNC, ASYNC):
406 "execution_mode should be None/SYNC/ASYNC. Got %s" % execution_mode)
407 if execution_mode is None:
408 execution_mode = SYNC
409 self._default_is_async = execution_mode == ASYNC
798 def execution_mode(self): member in Context
802 @execution_mode.setter
803 def execution_mode(self, mode): member in Context
1915 context().execution_mode = mode
[all …]
Dcore_test.py334 self.assertEqual(context.SYNC, ctx.execution_mode)
335 ctx.execution_mode = context.ASYNC
336 self.assertEqual(context.ASYNC, ctx.execution_mode)
337 ctx.execution_mode = context.SYNC
338 self.assertEqual(context.SYNC, ctx.execution_mode)
386 ctx = context.Context(execution_mode=context.ASYNC)
540 with context.execution_mode(context.ASYNC):
620 with context.execution_mode(context.ASYNC):
640 context.context().execution_mode = context.SYNC
994 ctx.execution_mode = mode
[all …]
/external/libchrome/base/task_scheduler/
Dscheduler_worker_pool_unittest.cc52 test::ExecutionMode execution_mode; member
64 test::ExecutionMode execution_mode, in ThreadPostingTasks() argument
70 execution_mode), in ThreadPostingTasks()
71 execution_mode) { in ThreadPostingTasks()
174 worker_pool_.get(), GetParam().execution_mode, PostNestedTask::NO)); in TEST_P()
196 worker_pool_.get(), GetParam().execution_mode, PostNestedTask::YES)); in TEST_P()
215 worker_pool_.get(), GetParam().execution_mode); in TEST_P()
225 worker_pool_.get(), GetParam().execution_mode); in TEST_P()
241 worker_pool_.get(), GetParam().execution_mode); in TEST_P()
275 worker_pool_.get(), GetParam().execution_mode); in TEST_P()
Dtest_utils.cc27 test::ExecutionMode execution_mode) { in CreateTaskRunnerWithExecutionMode() argument
30 switch (execution_mode) { in CreateTaskRunnerWithExecutionMode()
Dtest_utils.h46 test::ExecutionMode execution_mode);
Dtask_tracker.cc41 const char* execution_mode, in TaskTracingInfo() argument
44 execution_mode_(execution_mode), in TaskTracingInfo()
508 const char* const execution_mode = in RunOrSkipTask() local
518 task.traits, execution_mode, sequence_token)); in RunOrSkipTask()
/external/mesa3d/prebuilt-intermediates/nir/
Dnir_constant_expressions.c566 UNUSED unsigned execution_mode) in evaluate_amul() argument
675 UNUSED unsigned execution_mode) in evaluate_b16all_fequal16() argument
844 UNUSED unsigned execution_mode) in evaluate_b16all_fequal2() argument
1013 UNUSED unsigned execution_mode) in evaluate_b16all_fequal3() argument
1182 UNUSED unsigned execution_mode) in evaluate_b16all_fequal4() argument
1351 UNUSED unsigned execution_mode) in evaluate_b16all_fequal8() argument
1520 UNUSED unsigned execution_mode) in evaluate_b16all_iequal16() argument
1825 UNUSED unsigned execution_mode) in evaluate_b16all_iequal2() argument
2102 UNUSED unsigned execution_mode) in evaluate_b16all_iequal3() argument
2381 UNUSED unsigned execution_mode) in evaluate_b16all_iequal4() argument
[all …]
/external/mesa3d/src/compiler/nir/
Dnir_loop_analyze.c587 unsigned execution_mode) in eval_const_unop() argument
592 nir_eval_const_opcode(op, &dest, 1, bit_size, src, execution_mode); in eval_const_unop()
599 unsigned execution_mode) in eval_const_binop() argument
604 nir_eval_const_opcode(op, &dest, 1, bit_size, src, execution_mode); in eval_const_binop()
611 unsigned execution_mode) in get_iteration() argument
621 execution_mode); in get_iteration()
623 execution_mode); in get_iteration()
629 execution_mode); in get_iteration()
631 execution_mode); in get_iteration()
639 execution_mode); in get_iteration()
[all …]
Dnir.h1197 nir_is_float_control_signed_zero_inf_nan_preserve(unsigned execution_mode, unsigned bit_size) in nir_is_float_control_signed_zero_inf_nan_preserve() argument
1199 return (16 == bit_size && execution_mode & FLOAT_CONTROLS_SIGNED_ZERO_INF_NAN_PRESERVE_FP16) || in nir_is_float_control_signed_zero_inf_nan_preserve()
1200 (32 == bit_size && execution_mode & FLOAT_CONTROLS_SIGNED_ZERO_INF_NAN_PRESERVE_FP32) || in nir_is_float_control_signed_zero_inf_nan_preserve()
1201 (64 == bit_size && execution_mode & FLOAT_CONTROLS_SIGNED_ZERO_INF_NAN_PRESERVE_FP64); in nir_is_float_control_signed_zero_inf_nan_preserve()
1205 nir_is_denorm_flush_to_zero(unsigned execution_mode, unsigned bit_size) in nir_is_denorm_flush_to_zero() argument
1207 return (16 == bit_size && execution_mode & FLOAT_CONTROLS_DENORM_FLUSH_TO_ZERO_FP16) || in nir_is_denorm_flush_to_zero()
1208 (32 == bit_size && execution_mode & FLOAT_CONTROLS_DENORM_FLUSH_TO_ZERO_FP32) || in nir_is_denorm_flush_to_zero()
1209 (64 == bit_size && execution_mode & FLOAT_CONTROLS_DENORM_FLUSH_TO_ZERO_FP64); in nir_is_denorm_flush_to_zero()
1213 nir_is_denorm_preserve(unsigned execution_mode, unsigned bit_size) in nir_is_denorm_preserve() argument
1215 return (16 == bit_size && execution_mode & FLOAT_CONTROLS_DENORM_PRESERVE_FP16) || in nir_is_denorm_preserve()
[all …]
Dnir_search.c877 const unsigned execution_mode = in nir_algebraic_instr() local
880 nir_is_float_control_signed_zero_inf_nan_preserve(execution_mode, bit_size) || in nir_algebraic_instr()
881 nir_is_denorm_flush_to_zero(execution_mode, bit_size); in nir_algebraic_instr()
/external/tensorflow/tensorflow/python/framework/
Dconfig.py275 return context.context().execution_mode == context.SYNC
296 context.context().execution_mode = None
298 context.context().execution_mode = context.SYNC
300 context.context().execution_mode = context.ASYNC
Dgraph_building_benchmark.py66 with context.execution_mode(context.GRAPH_MODE):
76 with context.execution_mode(context.GRAPH_MODE):
88 with context.execution_mode(context.GRAPH_MODE):
Dconfig_test.py110 self.assertEqual(context.SYNC, context.context().execution_mode)
120 self.assertEqual(context.SYNC, context.context().execution_mode)
123 self.assertEqual(context.ASYNC, context.context().execution_mode)
/external/mesa3d/src/compiler/spirv/
Dvtn_glsl450.c246 unsigned execution_mode, in vtn_nir_alu_op_for_spirv_glsl_opcode() argument
295 if (execution_mode & FLOAT_CONTROLS_DENORM_FLUSH_TO_ZERO_FP16) in vtn_nir_alu_op_for_spirv_glsl_opcode()
549 unsigned execution_mode = in handle_glsl450_alu() local
552 nir_op op = vtn_nir_alu_op_for_spirv_glsl_opcode(b, entrypoint, execution_mode, &exact); in handle_glsl450_alu()
Dspirv_to_nir.c4787 unsigned execution_mode = 0; in vtn_handle_execution_mode() local
4791 case 16: execution_mode = FLOAT_CONTROLS_DENORM_PRESERVE_FP16; break; in vtn_handle_execution_mode()
4792 case 32: execution_mode = FLOAT_CONTROLS_DENORM_PRESERVE_FP32; break; in vtn_handle_execution_mode()
4793 case 64: execution_mode = FLOAT_CONTROLS_DENORM_PRESERVE_FP64; break; in vtn_handle_execution_mode()
4799 case 16: execution_mode = FLOAT_CONTROLS_DENORM_FLUSH_TO_ZERO_FP16; break; in vtn_handle_execution_mode()
4800 case 32: execution_mode = FLOAT_CONTROLS_DENORM_FLUSH_TO_ZERO_FP32; break; in vtn_handle_execution_mode()
4801 case 64: execution_mode = FLOAT_CONTROLS_DENORM_FLUSH_TO_ZERO_FP64; break; in vtn_handle_execution_mode()
4807 case 16: execution_mode = FLOAT_CONTROLS_SIGNED_ZERO_INF_NAN_PRESERVE_FP16; break; in vtn_handle_execution_mode()
4808 case 32: execution_mode = FLOAT_CONTROLS_SIGNED_ZERO_INF_NAN_PRESERVE_FP32; break; in vtn_handle_execution_mode()
4809 case 64: execution_mode = FLOAT_CONTROLS_SIGNED_ZERO_INF_NAN_PRESERVE_FP64; break; in vtn_handle_execution_mode()
[all …]
/external/tensorflow/tensorflow/python/data/kernel_tests/
Ddataset_test.py380 combinations.combine(execution_mode=[context.ASYNC, context.SYNC])))
381 def testEagerIteration(self, execution_mode): argument
382 with context.execution_mode(execution_mode):
/external/tensorflow/tensorflow/python/training/tracking/
Dbenchmarks_test.py77 def _run(self, func, num_iters, execution_mode=None): argument
/external/tensorflow/tensorflow/python/distribute/
Dmirrored_function_strategy.py125 with context.execution_mode(context.ASYNC):
/external/mesa3d/src/intel/compiler/
Dbrw_fs_visitor.cpp254 unsigned execution_mode = this->nir->info.float_controls_execution_mode; in emit_shader_float_controls_execution_mode() local
255 if (execution_mode == FLOAT_CONTROLS_DEFAULT_FLOAT_CONTROL_MODE) in emit_shader_float_controls_execution_mode()
259 unsigned mask, mode = brw_rnd_mode_from_nir(execution_mode, &mask); in emit_shader_float_controls_execution_mode()
/external/swiftshader/third_party/SPIRV-Tools/source/val/
Dvalidation_state.h246 SpvExecutionMode execution_mode) { in RegisterExecutionModeForEntryPoint() argument
247 entry_point_to_execution_modes_[entry_point].insert(execution_mode); in RegisterExecutionModeForEntryPoint()
/external/deqp-deps/SPIRV-Tools/source/val/
Dvalidation_state.h246 SpvExecutionMode execution_mode) { in RegisterExecutionModeForEntryPoint() argument
247 entry_point_to_execution_modes_[entry_point].insert(execution_mode); in RegisterExecutionModeForEntryPoint()
/external/angle/third_party/spirv-tools/src/source/val/
Dvalidation_state.h246 SpvExecutionMode execution_mode) { in RegisterExecutionModeForEntryPoint() argument
247 entry_point_to_execution_modes_[entry_point].insert(execution_mode); in RegisterExecutionModeForEntryPoint()

12