Searched refs:np_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend_test.py | 40 np_op, argument
58 np_output = np_op(inputs.astype(dtype), *np_args, **np_kwargs)
68 np_op, argument
86 np_output = np_op(input_a.astype(dtype), input_b.astype(dtype),
309 for keras_op, np_op in ops_to_test:
310 compare_single_input_op_to_numpy(keras_op, np_op, input_shape=(4, 7, 5),
313 compare_single_input_op_to_numpy(keras_op, np_op, input_shape=(4, 7, 5),
317 compare_single_input_op_to_numpy(keras_op, np_op,
334 for keras_op, np_op in ops_to_test:
335 compare_single_input_op_to_numpy(keras_op, np_op, input_shape=(4, 7))
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_ops_test.py | 736 np_op = np.sum
739 np_op = np.max
744 np_ans = np_op(np_ans, keepdims=keep_dims)
754 np_ans = np_op(np_ans, axis=ra, keepdims=keep_dims)
|