Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/
H A Dlinear_operator_tridiag_test.py22 from tensorflow.python.ops import variables as variables_module unknown
125 diag = variables_module.Variable([[3., 6., 2.], [2., 4., 2.], [5., 1., 2.]])
131 diag = variables_module.Variable([[3., 6., 2.], [2., 4., 2.], [5., 1., 2.]])
164 variables_module.Variable([3., 6., 2.]),
165 variables_module.Variable([2., 4., 2.]),
166 variables_module.Variable([5., 1., 2.])]
175 variables_module.Variable([2., 4., 2.]),
206 matrix = variables_module.Variable([[3., 2., 0.], [1., 6., 4.], [0., 2, 2]])
H A Dlinear_operator_low_rank_update_test.py23 from tensorflow.python.ops import variables as variables_module unknown
158 variables_module.Variable([1.], name="diag"),
164 u=variables_module.Variable([[2.]], name="u"),
165 v=variables_module.Variable([[1.25]], name="v")
167 diag_update=variables_module.Variable([1.25], name="diag_update")
174 variables_module.Variable([1.], name="diag"),
180 u=variables_module.Variable([[2.]], name="u"),
181 v=variables_module.Variable([[1.25]], name="v")
183 diag_update=variables_module.Variable([1.25], name="diag_update")
H A Dlinear_operator_block_lower_triangular_test.py24 from tensorflow.python.ops import variables as variables_module unknown
184 variables_module.Variable([[1., 0.], [0., 1.]]),
188 variables_module.Variable([[2., 0.], [1., 0.]]))
190 variables_module.Variable([[3., 1.], [1., 3.]]),
211 variables_module.Variable([[1., 0.], [0., 1.]]),
215 variables_module.Variable([[2., 0.], [1., 0.]]))
217 variables_module.Variable([[3., 1.], [1., 3.]]),
H A Dlinear_operator_toeplitz_test.py26 from tensorflow.python.ops import variables as variables_module unknown
139 col = variables_module.Variable([1.])
140 row = variables_module.Variable([1.])
158 col = variables_module.Variable([1.])
159 row = variables_module.Variable([1.])
H A Dlinear_operator_full_matrix_test.py24 from tensorflow.python.ops import variables as variables_module unknown
113 matrix = variables_module.Variable([[2.]])
118 matrix = variables_module.Variable([[3.]])
218 matrix = variables_module.Variable([[2.]])
260 matrix = variables_module.Variable([[2., 1.]])
H A Dlinear_operator_kronecker_test.py23 from tensorflow.python.ops import variables as variables_module unknown
261 matrix_1 = variables_module.Variable([[1., 0.], [0., 1.]])
262 matrix_2 = variables_module.Variable([[2., 0.], [0., 2.]])
275 matrix_1 = variables_module.Variable([[1., 0.], [0., 1.]])
276 matrix_2 = variables_module.Variable([[2., 0.], [0., 2.]])
H A Dlinear_operator_zeros_test.py21 from tensorflow.python.ops import variables as variables_module unknown
192 linalg_lib.LinearOperatorZeros(num_rows=variables_module.Variable(2))
196 num_rows=2, num_columns=variables_module.Variable(3))
200 num_rows=2, batch_shape=variables_module.Variable([2]))
H A Dlinear_operator_householder_test.py21 from tensorflow.python.ops import variables as variables_module unknown
99 reflection_axis = variables_module.Variable([1., 3., 5., 8.])
112 reflection_axis = variables_module.Variable([1., 3., 5., 8.])
H A Dlinear_operator_lower_triangular_test.py20 from tensorflow.python.ops import variables as variables_module unknown
114 tril = variables_module.Variable([[1., 0.], [0., 1.]])
120 tril = variables_module.Variable([[1., 0.], [0., 1.]])
H A Dlinear_operator_identity_test.py25 from tensorflow.python.ops import variables as variables_module unknown
285 linalg_lib.LinearOperatorIdentity(num_rows=variables_module.Variable(2))
289 num_rows=2, batch_shape=variables_module.Variable([3]))
572 num_rows=variables_module.Variable(2), multiplier=1.23)
575 multiplier = variables_module.Variable(1.23)
581 multiplier = variables_module.Variable(1.23)
H A Dlinear_operator_diag_test.py22 from tensorflow.python.ops import variables as variables_module unknown
250 diag = variables_module.Variable([[2.]])
255 diag = variables_module.Variable([[2.]])
H A Dlinear_operator_inversion_test.py20 from tensorflow.python.ops import variables as variables_module unknown
134 matrix = variables_module.Variable([[1., 2.], [3., 4.]])
H A Dlinear_operator_circulant_test.py25 from tensorflow.python.ops import variables as variables_module unknown
338 spectrum = variables_module.Variable(
344 spectrum = variables_module.Variable(
418 spectrum = variables_module.Variable(
886 spectrum = variables_module.Variable(
H A Dlinear_operator_block_diag_test.py24 from tensorflow.python.ops import variables as variables_module unknown
343 matrices.append(variables_module.Variable(
359 matrices.append(variables_module.Variable(
H A Dlinear_operator_adjoint_test.py22 from tensorflow.python.ops import variables as variables_module unknown
244 matrix = variables_module.Variable([[1., 2.], [3., 4.]])
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/linalg/
H A Dlinear_operator_util.py27 from tensorflow.python.ops import variables as variables_module unknown
160 isinstance(x, variables_module.Variable) or
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dbackend.py75 from tensorflow.python.ops import variables as variables_module unknown
214 variables_module.Variable,
1086 v = variables_module.Variable(
1198 [variables_module.is_variable_initialized(v) for v in candidate_vars])
1210 session.run(variables_module.variables_initializer(uninitialized_vars))
1278 (ops.Tensor, variables_module.Variable,
4891 if (not isinstance(output, (ops.EagerTensor, variables_module.Variable)) and
4947 not isinstance(output, (ops.EagerTensor, variables_module.Variable)) and
5037 if (not isinstance(output, (ops.EagerTensor, variables_module.Variable)) and
6490 if isinstance(tensor, variables_module.Variable):
H A Dmetrics.py66 from tensorflow.python.ops import variables as variables_module unknown
291 aggregation=variables_module.VariableAggregation.SUM,
292 synchronization=variables_module.VariableSynchronization.ON_READ,
303 synchronization = variables_module.VariableSynchronization.ON_WRITE
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/utils/
H A Dmetrics_utils.py40 from tensorflow.python.ops import variables as variables_module unknown
151 (ops.Tensor, variables_module.Variable, float, int)):
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/saving/
H A Dhdf5_format.py30 from tensorflow.python.ops import variables as variables_module unknown
894 if any(not isinstance(w, variables_module.Variable) for w in weights):