OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:col_remapping
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/python/kernel_tests/
checkpoint_ops_test.py
139
col_remapping
=[],
149
col_remapping
= [2, 0]
154
col_remapping
=
col_remapping
,
157
num_cols=len(
col_remapping
))
159
self.assertAllClose(self.matrix_value[row_remapping][:,
col_remapping
],
164
col_remapping
= [1, 15]
169
col_remapping
=
col_remapping
,
172
num_cols=len(
col_remapping
))
[
all
...]
/external/tensorflow/tensorflow/core/kernels/
load_and_remap_matrix_op.cc
58
// swaps around the rows/columns according to row_remapping/
col_remapping
.
104
OP_REQUIRES_OK(context, context->input("
col_remapping
", &col_remapping_t));
105
const auto
col_remapping
= col_remapping_t->vec<int64>();
variable
109
const bool remap_cols =
col_remapping
.size() > 0;
112
context,
col_remapping
.size() == num_cols_,
114
"Provided
col_remapping
, but its size is ",
col_remapping
.size(),
116
OP_REQUIRES_OK(context, RemapVectorToMap(
col_remapping
, &col_id_present,
/external/tensorflow/tensorflow/python/training/
checkpoint_ops.py
167
col_remapping
= []
170
col_remapping
, num_cols_present = (
185
col_remapping
=
col_remapping
,
Completed in 147 milliseconds