HomeSort by relevance Sort by last modified time
    Searched refs:saved_shape (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
save_restore_tensor.cc 174 TensorShape saved_shape; local
176 context, reader->HasTensor(tensor_name, &saved_shape, &type),
187 TensorShape output_shape(saved_shape);
188 TensorSlice slice_to_load(saved_shape.dims());
197 context, parsed_shape.IsSameSize(saved_shape),
202 ", save file shape: ", saved_shape.DebugString()));
  /external/tensorflow/tensorflow/core/util/
tensor_slice_writer.h 145 TensorShape saved_shape(ssm->shape());
147 TF_RETURN_IF_ERROR(slice.SliceTensorShape(saved_shape, &sliced_shape));

Completed in 410 milliseconds