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"
|