Home
last modified time | relevance | path

Searched refs:saved_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_tensor.cc179 TensorShape saved_shape; in RestoreTensor() local
181 context, reader->HasTensor(tensor_name, &saved_shape, &type), in RestoreTensor()
192 TensorShape output_shape(saved_shape); in RestoreTensor()
193 TensorSlice slice_to_load(saved_shape.dims()); in RestoreTensor()
203 context, parsed_shape.IsSameSize(saved_shape), in RestoreTensor()
208 ", save file shape: ", saved_shape.DebugString())); in RestoreTensor()
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_writer.h144 TensorShape saved_shape(ssm->shape()); in Add()
146 TF_RETURN_IF_ERROR(slice.SliceTensorShape(saved_shape, &sliced_shape)); in Add()