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.cc353 const TensorShape& full_tensor_shape) { in IsFullSlice() argument
358 slice_spec.SliceTensorShape(full_tensor_shape, &sliced_shape).IgnoreError(); in IsFullSlice()
359 return sliced_shape == full_tensor_shape; in IsFullSlice()
477 const TensorShape& full_tensor_shape, in AddSlice() argument
484 if (IsFullSlice(slice_spec, full_tensor_shape)) { in AddSlice()
499 CHECK(TensorShape(full_entry->shape()) == full_tensor_shape); in AddSlice()
505 full_tensor_shape.AsProto(full_entry->mutable_shape()); in AddSlice()
Dtensor_bundle.h140 const TensorShape& full_tensor_shape,