Searched refs:shape_key (Results 1 – 4 of 4) sorted by relevance
287 shape_key=None, argument310 self._shape_key = shape_key315 if shape_key:316 keys.append(shape_key)
547 'labels': tfexample_decoder.SparseTensor(shape_key='shape'),
892 shape_key = layer.name + '_0_0'893 layers_to_output_shapes[shape_key] = shape927 shape_key = layer.name + '_%s_%s' % (node_index, j)928 layers_to_output_shapes[shape_key] = shape934 shape_key = layer.name + '_%s_%s' % (node_index, tensor_index)935 output_shapes.append(layers_to_output_shapes[shape_key])
1179 shape_key = tuple([tuple(spec.shape.as_list()) for spec in input_specs])1180 if shape_key not in self._compilation_cache:1187 self._compilation_cache[shape_key] = new_tpu_model_ops1190 return self._compilation_cache[shape_key]