| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2_cpu_test.py | 173 row_starts = ragged.row_starts().numpy()
174 reduced.append(np.add.reduceat(all_lookups, row_starts))
199 row_starts = ragged_tensor.RaggedTensor.from_sparse(feature).row_starts()
200 row_starts = row_starts.numpy()
201 weighted_sum.append(np.add.reduceat(all_lookups, row_starts))
203 weighted_sum[-1] /= np.add.reduceat(weight, row_starts)
248 row_starts = feature.row_starts().numpy()
249 weighted_sum.append(np.add.reduceat(all_lookups, row_starts))
251 weighted_sum[-1] /= np.add.reduceat(weight, row_starts)
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | row_partition.py | 402 row_starts,
425 with ops.name_scope(None, "RowPartitionFromRowStarts", [row_starts]):
426 row_starts = cls._convert_row_partition(row_starts, "row_starts",
428 row_starts.shape.assert_has_rank(1)
429 nvals = math_ops.cast(nvals, row_starts.dtype)
433 check_ops.assert_rank(row_starts, 1, message=msg),
434 _assert_zero(row_starts[:1], message=msg),
435 _assert_monotonic_increasing(row_starts, message=msg),
436 check_ops.assert_less_equal(row_starts[-1:], nvals, message=msg),
438 row_starts = control_flow_ops.with_dependencies(checks, row_starts)
[all …]
|
| D | ragged_tensor_supported_values_test.py | 165 row_starts = constant_op.constant([0, 2, 2, 5, 6], dtypes.int64)
166 rt = RaggedTensor.from_row_starts(values, row_starts)
169 self.assertAllEqual(rt.row_starts(), row_starts)
|
| D | row_partition_test.py | 55 rt4 = RowPartition.from_row_starts(row_starts=[0, 4, 4, 7, 8], nvals=8)
209 row_starts = constant_op.constant([0, 2, 2, 5, 6], dtypes.int64)
211 rp = RowPartition.from_row_starts(row_starts, nvals, validate=False)
214 rp_row_starts = rp.row_starts()
219 self.assertAllEqual(rp_row_starts, row_starts)
|
| D | ragged_gather_ops.py | 310 return math_ops.cast(t.row_starts(), dtype)
461 flattened_params.row_starts(), flattened_index_tuples)
|
| D | ragged_tensor_test.py | 81 rt4 = RaggedTensor.from_row_starts(values, row_starts=[0, 4, 4, 7, 8])
326 row_starts = constant_op.constant([0, 2, 2, 5, 6], dtypes.int64)
328 rt = RaggedTensor.from_row_starts(values, row_starts, validate=False)
334 rt_row_starts = rt.row_starts()
339 self.assertAllEqual(rt_row_starts, row_starts)
667 self.assertAllEqual(rt.row_starts(), [0, 2, 2, 5, 6])
694 self.assertAllEqual(rt.row_starts(), [0, 2, 2, 5, 6])
730 self.assertAllEqual(rt.row_starts(), [0, 2, 3, 3])
965 'test': RaggedTensor.row_starts,
|
| D | ragged_tensor.py | 497 def from_row_starts(cls, values, row_starts, name=None, validate=True): argument
526 with ops.name_scope(name, "RaggedFromRowStarts", [values, row_starts]):
529 row_starts=row_starts,
1126 def row_starts(self, name=None): member in RaggedTensor
1149 return self._row_partition.row_starts()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | scale_and_translate_op.cc | 358 typename TTypes<int32, 1>::ConstTensor row_starts( in Compute() local
369 context->eigen_device<Device>(), row_spans.span_size, row_starts, in Compute()
463 typename TTypes<int32, 1>::ConstTensor row_starts = in Compute() local
474 context->eigen_device<Device>(), row_spans.span_size, row_starts, in Compute()
558 typename TTypes<int32, 1>::ConstTensor row_starts, in operator ()()
585 GatherRows(row_span_size, row_starts.data(), row_weights.data(), in operator ()()
|
| D | scale_and_translate_op.h | 62 typename TTypes<int32, 1>::ConstTensor row_starts,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-ragged-tensor.pbtxt | 77 …argspec: "args=[\'cls\', \'values\', \'row_starts\', \'name\', \'validate\'], varargs=None, keywor…
128 name: "row_starts"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-ragged-tensor.pbtxt | 77 …argspec: "args=[\'cls\', \'values\', \'row_starts\', \'name\', \'validate\'], varargs=None, keywor…
128 name: "row_starts"
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | ragged_keras_tensor_test.py | 229 inp, row_starts=[0, 4, 4, 7, 8])
234 x, row_starts=[0, 4, 4, 7, 8])
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | parsing_config.py | 852 rt.row_starts(), partition_t.row_lengths())
858 rt.row_starts(), partition_t.row_lengths())
|