Home
last modified time | relevance | path

Searched refs:_reduce (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/
H A Dcross_device_ops_test.py550 def _reduce(test_object, reduce_op, per_replica, destinations): function
592 _reduce(
599 _reduce(
606 _reduce(
614 _reduce(
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/
H A Dnp_array_ops.py469 def _reduce(tf_fn, function
556 return _reduce(
567 return _reduce(
580 return _reduce(
593 return _reduce(
608 return _reduce(
654 result = _reduce(
668 return _reduce(
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_lib.py1427 return self._extended._reduce(reduce_op, value) # pylint: disable=protected-access
1448 return self._extended._reduce(reduce_op, value) # pylint: disable=protected-access
1516 …numer = self._extended._reduce(reduce_util.ReduceOp.SUM, numer) # pylint: disable=protected-access
1517 …denom = self._extended._reduce(reduce_util.ReduceOp.SUM, denom) # pylint: disable=protected-access
2296 def _reduce(self, reduce_op, value): member in StrategyExtendedV2
H A Dinput_lib_test.py1169 def _reduce(state, batch): function
1176 sums = _reduce(sums, batch)
1189 defun(lambda state, iterator: _reduce(state, next(iterator))))
1310 def _reduce(state, batch): function
1322 sums = _reduce(sums, opt_iterate.get_value())
H A Dparameter_server_strategy_v2.py921 def _reduce(self, reduce_op, value): member in ParameterServerStrategyV2Extended
/aosp_15_r20/external/pytorch/torch/jit/
H A D_script.py70 def _reduce(cls): function
74 ScriptFunction.__reduce__ = _reduce # type: ignore[assignment]
H A D_script.pyi60 def _reduce(cls) -> None: ...
/aosp_15_r20/external/capstone/bindings/python/pyx/
H A Dccapstone.pyx356 all_archs += "_reduce"
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dtraining.py2825 def _reduce(v): function
2841 return nest.map_structure(_reduce, values)
/aosp_15_r20/external/capstone/
H A DMakefile183 X86_REDUCE = _reduce
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_control_flow.py254 def _reduce(self, *operands): member in ReduceMod
258 return self._reduce(*operands)