Searched refs:keras_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend_test.py | 39 def compare_single_input_op_to_numpy(keras_op, argument
55 keras_output = keras_op(keras.backend.variable(inputs, dtype=dtype),
62 raise AssertionError('Test for op `' + str(keras_op.__name__) + '` failed; '
67 def compare_two_inputs_op_to_numpy(keras_op, argument
82 keras_output = keras_op(keras.backend.variable(input_a, dtype=dtype),
91 raise AssertionError('Test for op `' + str(keras_op.__name__) + '` failed; '
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),
316 if 'keepdims' in tf_inspect.getargspec(keras_op).args:
[all …]
|