Searched refs:col_remapping (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | checkpoint_ops_test.py | 141 col_remapping=[], 151 col_remapping = [2, 0] 156 col_remapping=col_remapping, 159 num_cols=len(col_remapping)) 161 self.assertAllClose(self.matrix_value[row_remapping][:, col_remapping], 166 col_remapping = [1, 15] 171 col_remapping=col_remapping, 174 num_cols=len(col_remapping)) 176 self.assertAllClose(self.matrix_value[row_remapping][:, col_remapping], 186 col_remapping=[1, -1], [all …]
|
/external/tensorflow/tensorflow/python/training/ |
D | checkpoint_ops.py | 167 col_remapping = [] 170 col_remapping, num_cols_present = ( 185 col_remapping=col_remapping,
|
/external/tensorflow/tensorflow/core/api_def/base_api/ |
D | api_def_LoadAndRemapMatrix.pbtxt | 26 name: "col_remapping" 82 * `col_remapping` must have either 0 entries (indicating that no column 85 `col_remapping[j]` in the old `Tensor` from the checkpoint. 89 `col_remapping` has `c` missing entries, then the following condition must be 96 As an example, with row_remapping = [1, 0, -1], col_remapping = [0, 2, -1],
|
/external/tensorflow/tensorflow/core/kernels/ |
D | load_and_remap_matrix_op.cc | 105 const auto col_remapping = col_remapping_t->vec<int64>(); in Compute() local 109 const bool remap_cols = col_remapping.size() > 0; in Compute() 112 context, col_remapping.size() == num_cols_, in Compute() 114 "Provided col_remapping, but its size is ", col_remapping.size(), in Compute() 116 OP_REQUIRES_OK(context, RemapVectorToMap(col_remapping, &col_id_present, in Compute()
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.raw_ops.pbtxt | 1693 …argspec: "args=[\'ckpt_path\', \'old_tensor_name\', \'row_remapping\', \'col_remapping\', \'initia…
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.raw_ops.pbtxt | 1693 …argspec: "args=[\'ckpt_path\', \'old_tensor_name\', \'row_remapping\', \'col_remapping\', \'initia…
|
/external/tensorflow/tensorflow/go/op/ |
D | wrappers.go | 36216 …t_path tf.Output, old_tensor_name tf.Output, row_remapping tf.Output, col_remapping tf.Output, ini… 36227 ckpt_path, old_tensor_name, row_remapping, col_remapping, initializing_values,
|
/external/tensorflow/tensorflow/core/ops/ |
D | ops.pbtxt | 16531 name: "col_remapping"
|
/external/tensorflow/tensorflow/core/ops/compat/ |
D | ops_history.v2.pbtxt | 32531 name: "col_remapping"
|
D | ops_history.v1.pbtxt | 32916 name: "col_remapping"
|