Home
last modified time | relevance | path

Searched refs:convert_to_int_tensor (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_util.py57 convert_to_int_tensor = array_ops.convert_to_int_tensor variable
H A Dsegment_id_ops.py111 segment_ids = ragged_util.convert_to_int_tensor(segment_ids, "segment_ids",
115 num_segments = ragged_util.convert_to_int_tensor(num_segments,
H A Dragged_tensor_shape.py316 lengths = ragged_util.convert_to_int_tensor(
H A Dragged_array_ops.py244 multiples = ragged_util.convert_to_int_tensor(
H A Dragged_tensor.py1758 lengths = ragged_util.convert_to_int_tensor(lengths, "lengths",
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Darray_ops.py6698 def convert_to_int_tensor(tensor, name, dtype=dtypes.int32): function
6806 repeats = convert_to_int_tensor(repeats, name="repeats")
6808 repeats = convert_to_int_tensor(repeats, name="repeats", dtype=None)
H A Dmath_ops.py176 num_int = array_ops.convert_to_int_tensor(num, name="num")