Home
last modified time | relevance | path

Searched refs:_to_numpy (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dgradient_checker_v2.py64 def _to_numpy(a): function
121 xs_data = [_to_numpy(a) for a in xs_data]
172 grad = _to_numpy(grad_fn(dy_data, *xs)[0])
188 grad = _to_numpy(grad_fn(dy_data, *xs)[0])
230 xs = [np.asarray(_to_numpy(x)) for x in xs]
245 y_pos = _to_numpy(f(*xs))
248 y_neg = _to_numpy(f(*xs))
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dverification.py101 def _to_numpy(elem) -> list | np.ndarray: function
108 return [_to_numpy(inp) for inp in elem]
114 flattened.extend([_to_numpy(k), _to_numpy(elem[k])])
133 return [_to_numpy(v) for v in value_unpacked]
144 ort_inputs[input_name] = _to_numpy(input)
145 inputs = _to_numpy(inputs)
1122 _to_numpy(inputs),
1123 _to_numpy(outputs),
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dcontrol_flow_ops_py_test.py1292 self.assertAllEqual(gradient_checker_v2._to_numpy(grad_value), [1.0, 1.0])
1295 self.assertAllEqual(gradient_checker_v2._to_numpy(grad_value), [0.0, 2.0])
1314 self.assertAllEqual(gradient_checker_v2._to_numpy(grad_val), [[0., 0.],
1348 self.assertAllEqual(gradient_checker_v2._to_numpy(var_grad), [[0., 0.],
1353 self.assertAllEqual(gradient_checker_v2._to_numpy(x1_grad), [[0., 0., 0.],
1357 self.assertEqual(gradient_checker_v2._to_numpy(x2_grad), 6.)
3176 self.assertAllEqual(gradient_checker_v2._to_numpy(grad), [0., 2., 0., 2.])
3199 self.assertAllEqual(gradient_checker_v2._to_numpy(grad), [0., 2., 0., 2.])
3224 self.assertAllEqual(gradient_checker_v2._to_numpy(grad), [0., 6., 6., 0.])
3248 self.assertAllEqual(gradient_checker_v2._to_numpy(grads[0]), [0., 2., 0.])
[all …]