Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_tensor_dense_matmul_op_test.py325 def _timer(sess, ops_fn, iterations): argument
327 sess.run(ops_fn(10, sess))
331 sess.run(ops_fn(iterations, sess))
345 ops_fn = _sparse_tensor_dense_vs_dense_matmul_benchmark_dense(
351 ops_fn = _sparse_tensor_dense_vs_dense_matmul_benchmark_dense(
353 delta_dense = _timer(sess, ops_fn, 200)
363 ops_fn = _sparse_tensor_dense_vs_dense_matmul_benchmark_sparse(
371 ops_fn = _sparse_tensor_dense_vs_dense_matmul_benchmark_sparse(
373 delta_sparse = _timer(sess, ops_fn, 200)