HomeSort by relevance Sort by last modified time
    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