Searched refs:convert_to_int_tensor (Results 1 – 5 of 5) sorted by relevance
90 segment_ids = ragged_util.convert_to_int_tensor(segment_ids, "segment_ids")93 num_segments = ragged_util.convert_to_int_tensor(num_segments,
104 ragged_util.convert_to_int_tensor(107 inner_dim_sizes = ragged_util.convert_to_int_tensor(148 ragged_util.convert_to_int_tensor(292 lengths = ragged_util.convert_to_int_tensor(
32 def convert_to_int_tensor(tensor, name, dtype=dtypes.int32): function144 repeats = convert_to_int_tensor(repeats, name="repeats")
257 multiples = ragged_util.convert_to_int_tensor(
1200 lengths = ragged_util.convert_to_int_tensor(lengths, "lengths",