Searched refs:tensor_map_stack_keys (Results 1 – 2 of 2) sorted by relevance
50 def tensor_map_stack_keys(input_handle, key_dtype): function51 return gen_map_ops.tensor_map_stack_keys(input_handle, key_dtype)
147 keys = map_ops.tensor_map_stack_keys(m, k.dtype)153 keys = map_ops.tensor_map_stack_keys(m, k.dtype)163 keys = map_ops.tensor_map_stack_keys(m, dtypes.float32)173 keys = map_ops.tensor_map_stack_keys(m, dtypes.float32)186 keys = map_ops.tensor_map_stack_keys(m, dtypes.float32)