Searched refs:sliced_shape (Results 1 – 3 of 3) sorted by relevance
145 TensorShape sliced_shape; in Add() local146 TF_RETURN_IF_ERROR(slice.SliceTensorShape(saved_shape, &sliced_shape)); in Add()147 TF_RETURN_IF_ERROR(SaveData(data, sliced_shape.num_elements(), ss)); in Add()
357 TensorShape sliced_shape; in IsFullSlice() local358 slice_spec.SliceTensorShape(full_tensor_shape, &sliced_shape).IgnoreError(); in IsFullSlice()359 return sliced_shape == full_tensor_shape; in IsFullSlice()
390 auto sliced_shape = rewriter.create<SliceOp>( in matchAndRewrite() local401 sliced_shape); in matchAndRewrite()