Searched refs:row_key (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_getitem.py | 128 row_key = key_list[0]
131 if row_key is Ellipsis:
137 if row_key is array_ops.newaxis:
145 if isinstance(row_key, slice):
146 sliced_rt_input = _slice_ragged_row_dimension(rt_input, row_key)
171 if int(row_key) >= len(starts):
172 raise IndexError("Row key {} out of bounds".format(row_key))
175 row = rt_input.values[starts[row_key]:limits[row_key]]
179 def _slice_ragged_row_dimension(rt_input, row_key): argument
189 if row_key.start is None and row_key.stop is None and row_key.step is None:
[all …]
|
| /external/tensorflow/tensorflow/contrib/bigtable/kernels/test_kernels/ |
| D | bigtable_test_client_test.cc | 58 EXPECT_EQ((*itr)->row_key(), "r1"); in TEST()
85 EXPECT_EQ((*itr)->row_key(), "r1"); in TEST()
112 EXPECT_EQ((*itr)->row_key(), "r1"); in TEST()
122 EXPECT_EQ((*itr)->row_key(), "r2"); in TEST()
132 EXPECT_EQ((*itr)->row_key(), "r3"); in TEST()
159 EXPECT_EQ((*itr)->row_key(), "r1"); in TEST()
169 EXPECT_EQ((*itr)->row_key(), "r2"); in TEST()
179 EXPECT_EQ((*itr)->row_key(), "r3"); in TEST()
213 EXPECT_EQ((*itr)->row_key(), "r1"); in TEST()
223 EXPECT_EQ((*itr)->row_key(), "r2"); in TEST()
[all …]
|
| D | bigtable_test_client.h | 111 string row_key; member
|
| D | bigtable_test_client.cc | 317 auto* row = &table_.rows[string(request.row_key())]; in MutateRow()
371 auto* row = &table_.rows[string(i->row_key())]; in MutateRows()
|
| /external/tensorflow/tensorflow/contrib/bigtable/kernels/ |
| D | bigtable_sample_key_pairs_dataset_op.cc | 136 string row_key(row_keys[i].row_key); in Initialize() local
137 if (dataset()->key_range_.contains_key(row_key)) { in Initialize()
139 if (keys_.empty() && dataset()->key_range_.begin_key() != row_key) { in Initialize()
142 keys_.push_back(std::move(row_key)); in Initialize()
|
| D | bigtable_lookup_dataset_op.cc | 181 row_key_tensor.scalar<string>()() = string(row.row_key()); in ParseRow()
188 << row.row_key(); in ParseRow()
206 " not found in row: ", row.row_key()); in ParseRow()
|
| D | bigtable_scan_dataset_op.cc | 178 row_key_tensor.scalar<string>()() = string(row.row_key()); in ParseRow()
185 << row.row_key(); in ParseRow()
203 dataset()->columns_[i], " not found in row: ", row.row_key()); in ParseRow()
|
| D | bigtable_sample_keys_dataset_op.cc | 100 string(row_keys_[index_].row_key); in GetNextInternal()
|
| D | bigtable_prefix_key_dataset_op.cc | 101 output_tensor.scalar<string>()() = string(row.row_key()); in ParseRow()
|
| D | bigtable_range_key_dataset_op.cc | 107 output_tensor.scalar<string>()() = string(row.row_key()); in ParseRow()
|
| D | bigtable_kernels.cc | 278 << "): " << failure.mutation().row_key() in ComputeAsync()
|
| /external/tensorflow/tensorflow/contrib/bigtable/ |
| D | README.md | 62 for row_key in dataset:
63 print('Row key %d: %s' % (row_index, row_key))
|