Searched refs:use_static_shape_ (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/contrib/solvers/python/kernel_tests/ |
| D | util_test.py | 31 def _testCreateOperator(self, use_static_shape_): argument
37 if use_static_shape_:
49 if use_static_shape_:
66 def _testIdentityOperator(self, use_static_shape_): argument
72 if use_static_shape_:
84 if use_static_shape_:
|
| D | least_squares_test.py | 41 def _get_least_squares_tests(dtype_, use_static_shape_, shape_): argument
52 if use_static_shape_:
60 if use_static_shape_:
|
| D | lanczos_test.py | 42 def _get_lanczos_tests(dtype_, use_static_shape_, shape_, orthogonalize_, argument
52 if use_static_shape_:
68 if use_static_shape_:
|
| D | linear_equations_test.py | 41 def _get_linear_equations_tests(dtype_, use_static_shape_, shape_): argument
59 if use_static_shape_:
83 if use_static_shape_:
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | matmul_op_test.py | 70 def _GetMatMulTest(a_np_, b_np_, use_static_shape_, **kwargs_): argument
88 if use_static_shape_:
114 def _GetMatMulGradientTest(a_np_, b_np_, use_static_shape_, **kwargs_): argument
117 if not use_static_shape_ or a_np_.dtype in (np.int32, np.int64, np.float16):
|
| D | qr_op_test.py | 74 def _GetQrOpTest(dtype_, shape_, full_matrices_, use_static_shape_): argument
130 if use_static_shape_:
136 if use_static_shape_:
|
| D | norm_op_test.py | 67 def _GetNormOpTest(dtype_, shape_, ord_, axis_, keep_dims_, use_static_shape_): argument
72 if use_static_shape_:
|
| D | svd_op_test.py | 87 def _GetSvdOpTest(dtype_, shape_, use_static_shape_, compute_uv_, argument
150 if use_static_shape_:
158 if use_static_shape_:
166 if use_static_shape_:
|
| D | batch_matmul_op_test.py | 205 for use_static_shape_ in set([True, tf2.enabled()]):
207 "testBatchMatmulOp_" + name + ("_%s" % use_static_shape_),
209 use_static_shape_))
|