Searched refs:tensor_map_stack_keys (Results 1 – 2 of 2) sorted by relevance
46 def tensor_map_stack_keys(input_handle, key_dtype): function47 return gen_map_ops.tensor_map_stack_keys(input_handle, key_dtype)
143 keys = map_ops.tensor_map_stack_keys(m, k.dtype)149 keys = map_ops.tensor_map_stack_keys(m, k.dtype)159 keys = map_ops.tensor_map_stack_keys(m, dtypes.float32)169 keys = map_ops.tensor_map_stack_keys(m, dtypes.float32)182 keys = map_ops.tensor_map_stack_keys(m, dtypes.float32)