Searched refs:_to_numpy (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | gradient_checker_v2.py | 64 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 D | verification.py | 101 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 D | control_flow_ops_py_test.py | 1292 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 …]
|