Searched refs:transpose_input (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 676 def update_row_factors(self, sp_input=None, transpose_input=False): argument
707 True, sp_input=sp_input, transpose_input=transpose_input)
709 def update_col_factors(self, sp_input=None, transpose_input=False): argument
740 False, sp_input=sp_input, transpose_input=transpose_input)
744 transpose_input=False, argument
772 transpose_input=transpose_input,
777 transpose_input=False, argument
805 transpose_input=transpose_input,
811 transpose_input=False, argument
866 transpose_input = not transpose_input
[all …]
|
| D | factorization_ops_test.py | 150 transpose_input=False)
201 sp_input=sp_feeder, transpose_input=False)
218 transpose_input=False,
223 sp_input=sp_feeder, transpose_input=False)
273 sp_input=sp_feeder, transpose_input=False)
291 transpose_input=False,
296 sp_input=sp_feeder, transpose_input=False)
376 sp_input=sp_feeder, transpose_input=True)
393 transpose_input=True,
398 sp_input=sp_feeder, transpose_input=True)
[all …]
|
| D | wals.py | 283 lambda x: model.update_row_factors(sp_input=x, transpose_input=False),
288 lambda x: model.update_col_factors(sp_input=x, transpose_input=True),
326 transpose_input=False)
332 transpose_input=True)
353 sp_input=input_rows, transpose_input=False)
357 sp_input=input_cols, transpose_input=True)
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/ |
| D | imagenet_input.py | 79 transpose_input=False, argument
89 self.transpose_input = transpose_input
95 if self.transpose_input:
198 if self.transpose_input:
217 if self.transpose_input:
|
| D | main_estimator_tpu.py | 121 assert not FLAGS.transpose_input # channels_first only for GPU
124 if FLAGS.transpose_input and mode != tf.estimator.ModeKeys.PREDICT:
215 transpose_input=FLAGS.transpose_input,
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | hdf5_format.py | 505 def transpose_input(from_cudnn): function
563 kernels = transform_kernels(weights[0], transpose_input(from_cudnn),
599 kernels = transform_kernels(weights[0], transpose_input(from_cudnn),
|