Searched refs:range_tensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 5207 def _float_step_convert(range_tensor): argument
5208 if symbolic_helper._is_fp(range_tensor):
5209 range_tensor = g.op(
5211 g.op("Ceil", range_tensor),
5214 return range_tensor
5227 range_tensor = _float_step_convert(end)
5229 g, nonzero(g, ones(g, range_tensor, dtype, None, None)), [1]
5247 range_tensor = _float_step_convert(g.op("Div", g.op("Sub", end, start), step))
5249 g, nonzero(g, ones(g, range_tensor, None, None, None)), [1]
5263 range_tensor = _float_step_convert(g.op("Sub", end, start))
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | input.py | 311 range_tensor = math_ops.range(limit)
313 range_tensor, [], num_epochs, shuffle, seed, capacity,
|