Searched refs:restore_index (Results 1 – 2 of 2) sorted by relevance
146 int preferred_shard, bool restore_slice, int restore_index) { in RestoreTensor() argument159 const string& tensor_name = tensor_name_t.flat<string>()(restore_index); in RestoreTensor()185 context, type == context->expected_output_dtype(restore_index), in RestoreTensor()195 const string& shape_spec = context->input(2).flat<string>()(restore_index); in RestoreTensor()213 context->allocate_output(restore_index, output_shape, &t)); in RestoreTensor()
54 int preferred_shard, bool restore_slice, int restore_index);