Searched refs:sliced_shape (Results 1 – 2 of 2) sorted by relevance
146 TensorShape sliced_shape; in Add() local147 TF_RETURN_IF_ERROR(slice.SliceTensorShape(saved_shape, &sliced_shape)); in Add()148 TF_RETURN_IF_ERROR(SaveData(data, sliced_shape.num_elements(), ss)); in Add()
337 TensorShape sliced_shape; in IsFullSlice() local338 slice_spec.SliceTensorShape(full_tensor_shape, &sliced_shape).IgnoreError(); in IsFullSlice()339 return sliced_shape == full_tensor_shape; in IsFullSlice()