Searched refs:row_remapping (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | checkpoint_ops_test.py | 137 row_remapping = [1, 0]
141 row_remapping=row_remapping,
147 self.assertAllClose(self.matrix_value[row_remapping],
151 row_remapping = list(range(self.old_num_rows))
156 row_remapping=row_remapping,
159 num_rows=len(row_remapping),
162 self.assertAllClose(self.matrix_value[row_remapping][:, col_remapping],
166 row_remapping = [1, 0, 4]
171 row_remapping=row_remapping,
174 num_rows=len(row_remapping),
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | load_and_remap_matrix_op.cc | 77 const auto row_remapping = row_remapping_t->vec<int64>(); in Compute() local
78 OP_REQUIRES(context, row_remapping.size() == num_rows_, in Compute()
80 "Size of row_remapping is ", row_remapping.size(), in Compute()
82 OP_REQUIRES_OK(context, RemapVectorToMap(row_remapping, &row_id_present, in Compute()
89 for (int i = 0; i < row_remapping.size(); ++i) { in Compute()
91 (row_remapping(i) >= 0 && row_remapping(i) < min_old_row)) { in Compute()
92 min_old_row = row_remapping(i); in Compute()
95 (row_remapping(i) >= 0 && row_remapping(i) > max_old_row)) { in Compute()
96 max_old_row = row_remapping(i); in Compute()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_ops.py | 151 row_remapping, num_rows_present = (
162 row_remapping = math_ops.range(
184 row_remapping=row_remapping,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_LoadAndRemapMatrix.pbtxt | 17 name: "row_remapping"
79 * `row_remapping` must have exactly `num_rows` entries. Row `i` of the output
81 `row_remapping[i]` in the old `Tensor` from the checkpoint.
88 missing row or column. If `row_remapping` has `r` missing entries and
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 | 12 name: "row_remapping"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | LoadAndRemapMatrix.pbtxt | 12 name: "row_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 func LoadAndRemapMatrix(scope *Scope, ckpt_path tf.Output, old_tensor_name tf.Output, row_remapping…
26971 ckpt_path, old_tensor_name, row_remapping, col_remapping, initializing_values,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 21332 name: "row_remapping"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 31864 name: "row_remapping"
|