Searched refs:col_remapping (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | checkpoint_ops_test.py | 142 col_remapping=[],
152 col_remapping = [2, 0]
157 col_remapping=col_remapping,
160 num_cols=len(col_remapping))
162 self.assertAllClose(self.matrix_value[row_remapping][:, col_remapping],
167 col_remapping = [1, 15]
172 col_remapping=col_remapping,
175 num_cols=len(col_remapping))
177 self.assertAllClose(self.matrix_value[row_remapping][:, col_remapping],
187 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/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/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/ops/compat/ops_history_v2/ |
| D | LoadAndRemapMatrix.pbtxt | 16 name: "col_remapping"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | LoadAndRemapMatrix.pbtxt | 16 name: "col_remapping"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2133 …argspec: "args=[\'ckpt_path\', \'old_tensor_name\', \'row_remapping\', \'col_remapping\', \'initia…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2133 …argspec: "args=[\'ckpt_path\', \'old_tensor_name\', \'row_remapping\', \'col_remapping\', \'initia…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 26960 …t_path tf.Output, old_tensor_name tf.Output, row_remapping tf.Output, col_remapping tf.Output, ini…
26971 ckpt_path, old_tensor_name, row_remapping, col_remapping, initializing_values,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 21336 name: "col_remapping"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 31868 name: "col_remapping"
|