Home
last modified time | relevance | path

Searched refs:row_starts (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_batch_gather_ops.py105 params_starts = params.row_starts()
118 adjustments = array_ops.expand_dims(params.row_starts(), 1)
Dragged_tensor.py448 def from_row_starts(cls, values, row_starts, name=None): argument
472 with ops.name_scope(name, "RaggedFromRowStarts", [values, row_starts]):
474 row_starts = ops.convert_to_tensor(row_starts, dtypes.int64, "row_starts")
475 row_starts.shape.assert_has_rank(1)
477 row_splits = array_ops.concat([row_starts, nvals], axis=0)
831 def row_starts(self, name=None): member in RaggedTensor
Dragged_tensor_test.py137 rt4 = RaggedTensor.from_row_starts(values, row_starts=[0, 4, 4, 7, 8])
383 row_starts = constant_op.constant([0, 2, 2, 5, 6], dtypes.int64)
385 rt = RaggedTensor.from_row_starts(values, row_starts)
391 rt_row_starts = rt.row_starts()
396 self.assertAllEqual(rt_row_starts, row_starts)
616 self.assertAllEqual(rt.row_starts(), [0, 2, 2, 5, 6])
644 self.assertEqual(self.eval_to_list(rt.row_starts()), [0, 2, 2, 5, 6])
677 self.assertEqual(self.eval_to_list(rt.row_starts()), [0, 2, 3, 3])
Dragged_gather_ops.py253 flattened_params.row_starts(), flattened_index_tuples)
/external/tensorflow/tensorflow/core/kernels/
Dscale_and_translate_op.cc357 typename TTypes<int32, 1>::ConstTensor row_starts( in Compute() local
368 context->eigen_device<Device>(), row_spans.span_size, row_starts, in Compute()
462 typename TTypes<int32, 1>::ConstTensor row_starts = in Compute() local
473 context->eigen_device<Device>(), row_spans.span_size, row_starts, in Compute()
557 typename TTypes<int32, 1>::ConstTensor row_starts, in operator ()()
584 GatherRows(row_span_size, row_starts.data(), row_weights.data(), in operator ()()
Dscale_and_translate_op.h62 typename TTypes<int32, 1>::ConstTensor row_starts,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-ragged-tensor.pbtxt72 …argspec: "args=[\'cls\', \'values\', \'row_starts\', \'name\'], varargs=None, keywords=None, defau…
103 name: "row_starts"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-ragged-tensor.pbtxt72 …argspec: "args=[\'cls\', \'values\', \'row_starts\', \'name\'], varargs=None, keywords=None, defau…
103 name: "row_starts"