Searched refs:op_table (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/torch/distributed/_shard/ |
| D | op_registry_utils.py | 14 def _register_op(op, func, op_table): argument
26 op_table[op] = func
29 def _decorator_func(wrapped_func, op, op_table): argument
40 _register_op(op, wrapper, op_table)
|
| /external/mesa3d/src/gallium/drivers/i915/ |
| D | i915_fpc_optimize.c | 83 } op_table[TGSI_OPCODE_LAST] = { variable
130 return (op_table[opcode].num_dst > 0); in op_has_dst()
136 return op_table[opcode].num_dst; in op_num_dst()
142 return op_table[opcode].num_src; in op_num_src()
148 return op_table[opcode].commutes; in op_commutes()
168 return op_table[opcode].is_texture; in op_is_texture()
174 unsigned ne = op_table[opcode].neutral_element; in op_neutral_element()
|
| /external/pytorch/torch/distributed/_shard/sharded_tensor/ |
| D | __init__.py | 479 return functools.partial(_decorator_func, op=func, op_table=_CUSTOM_SHARDED_OPS)
486 return functools.partial(_decorator_func, op=func, op_table=_SHARDED_OPS)
|
| /external/pytorch/torch/distributed/_shard/sharding_spec/ |
| D | api.py | 136 _decorator_func, op=func, op_table=_CUSTOM_SHARDING_SPEC_OPS[class_name]
|
| /external/executorch/exir/emit/test/ |
| D | test_emit.py | 653 op_table = program.execution_plan[0].operators
659 op_table[
668 op_table[
683 op_table[
692 op_table[
701 op_table[
718 op_table[
|
| /external/python/cpython3/Lib/test/test_unittest/ |
| D | test_skipping.py | 86 op_table = ((unittest.skipUnless, False, True),
88 for deco, do_skip, dont_skip in op_table:
|