Searched refs:flat_b (Results 1 – 1 of 1) sorted by relevance
2033 def _cast_indexed_slice_indices(structure, flat_a, flat_b): argument2067 assert flat_b[index].dtype in (dtypes.int32, dtypes.int64)2068 if flat_a[index].dtype != flat_b[index].dtype:2069 if flat_b[index].dtype == dtypes.int32:2070 flat_b[index] = math_ops.cast(flat_b[index], dtypes.int64)