Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc333 const TensorShape& full_tensor_shape) { in IsFullSlice() argument
338 slice_spec.SliceTensorShape(full_tensor_shape, &sliced_shape).IgnoreError(); in IsFullSlice()
339 return sliced_shape == full_tensor_shape; in IsFullSlice()
450 const TensorShape& full_tensor_shape, in AddSlice() argument
457 if (IsFullSlice(slice_spec, full_tensor_shape)) { in AddSlice()
472 CHECK(TensorShape(full_entry->shape()) == full_tensor_shape); in AddSlice()
478 full_tensor_shape.AsProto(full_entry->mutable_shape()); in AddSlice()
Dtensor_bundle.h140 const TensorShape& full_tensor_shape,